args.py

import argparse
import field
import os
import config
from utils.log import logger as log

parser = argparse.ArgumentParser(description='logseq转正常markdown')
parser.add_argument('-f', '--file', dest='file', type=str, required=True, help='输入文件')
parser.add_argument('-o', '--output', dest='output', type=str, help='输出目录')
parser.add_argument('-e', '--env', dest='env', type=str, default=".env", required=False, help='配置文件')
parser.add_argument('-t', '--type', dest='type', type=str, default="hexo", required=False, help='输出格式')


def init():
    """
    初始化参数
    :return: args
    """
    log.info("正在初始化")
    arg = parser.parse_args()
    config.get_my_config(arg.env)
    # 处理文件后缀
    if arg.file[-3:] != '.md':
        log.debug("文件后缀不是.md，已自动添加")
        arg.file += '.md'
    if os.path.isfile(os.path.realpath(arg.file)):
        input_dir = os.path.dirname(arg.file)
    else:
        input_dir = config.myConfig['INPUT_DIR']
    output_dir = config.myConfig['OUTPUT_DIR']
    field.myField['output_dir']['value'] = output_dir if arg.output is None else os.path.realpath(arg.output)
    field.myField['input_file']['value'] = os.path.join(input_dir, os.path.basename(arg.file))
    field.myField['input_dir']['value'] = input_dir
    field.myField['output_file']['value'] = os.path.join(field.myField['output_dir']['value'],
                                                         os.path.basename(field.myField['input_file']['value']))
    field.myField['output_assets_dir']['value'] = os.path.join(field.myField['output_dir']['value'], '..',
                                                               "assets")
    log.info("参数初始化完成")
    return arg