更新版本升级工具

Author: JoeanAmier

README.md

@@ -179,9 +179,9 @@
 
 ## EXE 更新
 
-> **方案一：** 下载并解压文件，将旧版本的 `cache` 文件夹和 `settings.json` 文件复制到 `_internal` 文件夹。
+> **方案一：** 下载并解压文件，将旧版本的 `TikTokDownloader.db` 文件和 `settings.json` 文件复制到 `_internal` 文件夹。
 
-> **方案二：** 下载并解压文件，复制全部文件，直接覆盖旧版本文件。
+> **方案二：** 下载并解压文件（不要运行程序），复制全部文件，直接覆盖旧版本文件。
 
 # ⚠️ 免责声明\(Disclaimers\)
 

src/downloader/download.py

@@ -512,6 +512,7 @@ async def download_file(
                 RequestError,
                 StreamError,
         ) as e:
+            progress.remove_task(task_id)
             self.log.warning(f"{show} 下载中断，错误信息：{e}")
             self.delete_file(cache)
             await self.recorder.delete_id(id_)

src/tools/truncate.py

@@ -5,7 +5,7 @@ def is_chinese_char(char: str) -> bool:
     return 'CJK' in name(char, "")
 
 
-def truncate_string(s: str, length: int) -> str:
+def truncate_string(s: str, length: int = 64) -> str:
     count = 0
     result = ""
     for char in s:
@@ -19,12 +19,12 @@ def truncate_string(s: str, length: int) -> str:
     return result
 
 
-def trim_string(s: str, length: int) -> str:
+def trim_string(s: str, length: int = 64) -> str:
     length = length // 2 - 2
     return f"{s[:length]}...{s[-length:]}" if len(s) > length else s
 
 
-def beautify_string(s: str, length: int) -> str:
+def beautify_string(s: str, length: int = 64) -> str:
     count = 0
     for char in s:
         if is_chinese_char(char):  # 判断是否为中文字符

升级工具.py

@@ -1,22 +1,23 @@
-# 本程序仅用于 5.3 更新至 5.4
-# 已完成：更新配置文件参数
-# 待完成：迁移旧版下载记录
-
-
+from asyncio import run
 from json import dump
 from json import load
 from platform import system
 
 from src.config import Settings
 from src.custom import PROJECT_ROOT
+from src.manager import Database
 
 SETTING_ROOT = PROJECT_ROOT.joinpath("settings.json")
 ENCODE = "UTF-8-SIG" if system() == "Windows" else "UTF-8"
 
 
 def about():
-    print("本程序仅用于 5.3 更新至 5.4！")
+    print("本程序仅适用于 TikTokDownloader V5.3 更新至 V5.4！")
     print("请确保本程序放置于 main.py 或 main.exe 同级文件夹！")
+    print("本程序功能：")
+    print("更新配置文件参数")
+    print("迁移旧版映射数据")
+    print("迁移旧版下载记录")
 
 
 def update_params():
@@ -28,11 +29,37 @@ def update_params():
     print("已更新配置文件！")
 
 
-def main():
+async def update_map(db):
+    with PROJECT_ROOT.joinpath("cache/AccountCache.json").open("r+", encoding="utf-8") as f:
+        data = load(f)
+    for i, j in data.items():
+        await db.update_mapping_data(i, j["name"], j["mark"])
+        print("写入映射", i, j["name"], j["mark"])
+    print("已更新映射数据！")
+
+
+async def update_record(db):
+    with PROJECT_ROOT.joinpath("cache/IDRecorder.txt").open("r+", encoding="utf-8") as f:
+        data = {line.strip() for line in f}
+    for i in data:
+        await db.write_download_data(i)
+        print("写入记录", i)
+    print("已更新下载记录！")
+
+
+async def main():
     about()
+    if input("直接回车继续执行，输入任意内容结束运行："):
+        return
     if not input("即将更新配置文件，直接回车继续执行，输入任意内容跳过执行："):
         update_params()
+    async with Database() as db:
+        if not input("即将迁移映射数据，直接回车继续执行，输入任意内容跳过执行："):
+            await update_map(db)
+        if not input("即将迁移下载记录，直接回车继续执行，输入任意内容跳过执行："):
+            await update_record(db)
+    print("程序运行完成！")
 
 
 if __name__ == '__main__':
-    main()
+    run(main())