vndangkhoa/kv-tiktok-download
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

🏷修复一键下载视频bug

Browse source
This commit is contained in:
Evil0ctal 2022-04-11 23:36:41 -07:00 committed by GitHub
parent 4cb770bb76
commit 34882e87c7
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
Web/web_zh.py
View file

@ -105,12 +105,14 @@ def compress_file(tar_file, target_file):
if os.path.isfile(target_file): if os.path.isfile(target_file):
with tarfile.open(tar_file, 'w') as tar: with tarfile.open(tar_file, 'w') as tar:
tar.add(target_file) tar.add(target_file)
return 'finished'
else: else:
with tarfile.open(tar_file, 'w') as tar: with tarfile.open(tar_file, 'w') as tar:
for root, dirs, files in os.walk(target_file): for root, dirs, files in os.walk(target_file):
for single_file in files: for single_file in files:
filepath = os.path.join(root, single_file) filepath = os.path.join(root, single_file)
tar.add(filepath) tar.add(filepath)
return 'finished'
def clean_file(path): def clean_file(path):
@ -165,12 +167,13 @@ def video_download_window(result_dict):
output_path = save_path + '/output' output_path = save_path + '/output'
tarfile_name = download_time + '_total_' + str(total_amount) + '_videos.tar' tarfile_name = download_time + '_total_' + str(total_amount) + '_videos.tar'
output_file = output_path + '/' + tarfile_name output_file = output_path + '/' + tarfile_name
put_info('正在压缩视频文件，请勿关闭当前弹窗，完成后会在下方显示按钮...')
# 判断目录是否存在 # 判断目录是否存在
if not os.path.exists(output_path): if not os.path.exists(output_path):
os.mkdir(output_path) os.mkdir(output_path)
compress_file(tar_file=output_file, target_file=save_path) if compress_file(tar_file=output_file, target_file=save_path) == 'finished':
tar = open(output_file, "rb").read() tar = open(output_file, "rb").read()
put_file(tarfile_name, tar, '点击下载视频合集压缩包') put_file(tarfile_name, tar, '点击下载视频合集压缩包')
except Exception as e: except Exception as e:
print(str(e)) print(str(e))