main.py

import time
import shutil
import os
import argparse

from pathlib import Path

from extractor import Extractor
from applier import Applier
from processor import Processor
from repo_dump import Repo
from const import NEW_DICT_DIR, OLD_DICT_DIR
from update import update_dict
from logger import logger
from util import dict_update_splited_htmlContent


argparser = argparse.ArgumentParser()
argparser.add_argument(
    "--branch", type=str, default="dev", help="the specific branch name of the repo"
)
argparser.add_argument(
    "--pt-token",
    type=str,
    default="",
    help="paratranz token used to download dictionary, will be override by environment variale",
)
argparser.add_argument(
    "--no-update-repo",
    action="store_true",
    default=False,
    help="whether to update repo file",
)
argparser.add_argument(
    "--no-update-dict",
    action="store_true",
    default=False,
    help="whether to update dictionary file",
)
argparser.add_argument(
    "--special-process", action="store_true", help="whether to do special process"
)
argparser.add_argument(
    "--ignore-untranslated",
    action="store_true",
    help="whether to ignore untranslated entries",
)


def main():
    args = argparser.parse_args()

    branch = args.branch
    pt_token = (
        args.pt_token
        if os.environ.get("PARATRANZ_TOKEN") is None
        else os.environ.get("PARATRANZ_TOKEN")
    )
    if pt_token == "":
        pt_token = input(
            "请输入Paratranz的Acccess Token，\t或选择设置环境变量“PARATRANZ_TOKEN”/在main.py文件夹中修改--pt-token的default值："
        )

    new_dict_dir = Path(NEW_DICT_DIR)
    old_dict_dir = Path(OLD_DICT_DIR)

    logger.info("==== 正在移除临时文件 ====")
    shutil.rmtree(new_dict_dir, ignore_errors=True)
    shutil.rmtree(old_dict_dir, ignore_errors=True)

    repo = Repo(branch, pt_token)
    root = repo.source_dir

    if not args.no_update_repo:
        logger.info("==== 正在下载最新版本游戏源码 ====")
        repo.fetch_latest_version()
        logger.info("==== 正在解压最新版本游戏源码 ====")
        repo.unzip_latest_version()

    extractor = Extractor(root, new_dict_dir, repo.latest_commit)
    applier = Applier(root, new_dict_dir)
    processor = Processor(new_dict_dir, pt_token)

    logger.info("==== 正在提取翻译条目 ====")
    extractor.extract()

    if not args.no_update_dict:
        logger.info("==== 正在下载最新字典文件 ====")
        repo.fetch_latest_dict()
    logger.info("==== 正在解压最新字典文件 ====")
    repo.unzip_latest_dict()

    logger.info("==== 正在合并字典 ====")
    update_dict(old_dict_dir, new_dict_dir, args.ignore_untranslated)

    if args.special_process:
        logger.info("==== 正在应用特殊处理 ====")
        processor.process()

    logger.info("==== 正在应用字典 ====")
    applier.apply()


if __name__ == "__main__":
    start = time.time()
    main()
    end = time.time()
    logger.info("===== 总耗时 %ss =====", end - start)