run.py

# -*- coding: utf-8 -*-
# @Author  : Qliangw
# @Time    : 2022/3/1 21:05
# @Function: 程序入口

import argparse
import textwrap

from sync_data.app.sync import start_sync, init_database, merge_old_cfg
from sync_data.utils import log_detail
from sync_data.version import version as ver


def get_version():
    return ver


# 命令行参数
arg_parser = argparse.ArgumentParser(description="导入数据至notion数据库",
                                     usage='use "python %(prog)s --help" for more information',
                                     formatter_class=argparse.RawTextHelpFormatter)
arg_parser.add_argument('-m',
                        '--media',
                        help=textwrap.dedent('''\
                            输入模式
                                book    -- 书籍
                                    (例如：python run.py -m book -s do
                                movie   -- 影视
                                music   -- 音乐
                                game    -- 游戏'''))
arg_parser.add_argument('-s',
                        '--status',
                        help=textwrap.dedent('''\
                            输入媒体状态
                                wish    -- 想看/想读/想听
                                do      -- 在看/在读/在听
                                collect -- 看过/度过/听过
                                all     -- 全部'''))

arg_parser.add_argument('-f',
                        '--func',
                        help=textwrap.dedent('''\
                            功能参数
                                init     -- 初始化数据库
                                config   -- 合并旧配置'''))

arg_parser.add_argument('-v',
                        '--version',
                        action='version',
                        version=get_version(),
                        help='版本信息')


if __name__ == '__main__':

    args = arg_parser.parse_args()
    media = args.media
    status = args.status
    func = args.func

    if media in ['book', 'music', 'movie', 'game'] and status in ['do', 'wish', 'collect', 'all']:
        if func is not None:
            log_detail.warn("【Tip】参数过多，不做处理，退出程序")
            exit()
        else:
            if status == 'all':
                for i in ['do', 'wish', 'collect']:
                    log_detail.info("----开始拉取数据----")
                    log_detail.info(f"【RUN】开始获取{media}的{i}状态信息")
                    start_sync(media_type=media, media_status=i)
            else:
                log_detail.info("----开始拉取数据----")
                start_sync(media_type=media, media_status=status)
    elif media is not None or status is not None:
        log_detail.warn(f'【Tip】您输入的-m参数为< {media} >,请输入< python run.py -h >查看正确指令')
        log_detail.warn(f'【Tip】您输入的-s参数为< {status} >,请输入< python run.py -h >查看正确指令')
    elif func == 'init':
        init_database()
    elif func == "config":
        merge_old_cfg()
    elif func is not None:
        log_detail.warn(f'【Tip】您输入的-f参数为< {func} >，请输入< python run.py -h >查看正确指令')
    else:
        log_detail.warn(f'【Tip】请输入< python run.py -h >查看正确指令')

    log_detail.info("【RUN】程序结束！")