def config(data_file_path): """将传来的章节名,数据名,数据文件类型,数据文件路径保存到配置文件中,配置文件命名为common.ini""" data_path = os.path.dirname(data_file_path) # 数据文件目录 section_name = (os.path.splitext(data_file_path)[0]).split(os.path.sep)[-1] # 章节名 data_name = (os.path.splitext(data_file_path)[0]).split(os.path.sep)[-1] # 数据文件名字 data_type = os.path.splitext(data_file_path)[1].strip('.') # 数据文件类型 log = mylog.my_logger(base_path + '\\log\\' + 'confi\ g_log.log', logging.DEBUG, 'pengtaolog') config_path = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) config_path = os.path.join(config_path + r'\conf', 'common.ini') # 配置文件名路径 try: # 尝试打开配置文件并添加个人配置,若无配置文件这生成 config = configparser.ConfigParser() config.read(config_path) # 打开配置文件 config.add_section(section_name) # 添加用户个人章节 items_list = {'data_name': data_name, 'data\ _type': data_type, 'data_path': data_path} # 配置项 for k, v in items_list.items(): # 写入用户个人配置 config.set(section_name, k, v) except Exception: # 生成配置文件 config = configparser.ConfigParser() config.add_section(section_name) items_list = {'data_name': data_name, 'data_\ type': data_type, 'data_path': data_path} for k, v in items_list.items(): config.set(section_name, k, v) with open(config_path, 'w') as f: # 写入个人配置信息并保存配置文件 config.write(f) log.debug('配置信息已存入配置文件')
def main(): log = mylog.my_logger(BASE_PATH + '\\log\\' + 'execute\ _log.log', logging.DEBUG, 'pengtaolog') # 打开日志 data_file_path = login.register(log) # 用户登录,返回该用户数据文件名 log.info('用户登录,返回该用户数据文件名') memo_o = memo_out.MemoOut(data_file_path) # 生成Memo_Out对象 log.info('利用数据文件名进行操作') if len(sys.argv) >= 2: if sys.argv[1] in ['-se', '--send_email']: # 系统命令行参数调用发邮件 log.debug('系统命令行参数调用发邮件') while True: try: ask = input('输出整年(year),整月(month)') if ask == 'year': data_path, choose = memo_o.out_all_a_year() log.debug('获得json文件的路径,和用户年份选择') if ask == 'month': data_path, choose = memo_o.out_all_a_month() log.debug('获得json文件的路径,和用户月份选择') break except Exception: print('输入错误') send_email.send_email_attach(f'这是{choose}的数据,请查看!', data_path, 'lv\ [email protected]') # 发送给特定人物的邮件 if len(sys.argv) >= 2: if sys.argv[1] in ['-rj', '--return_json']: # 系统命令行参数调用返回json数据 log.debug('系统命令行参数调用返回json数据') print(memo_o.get_all_a_month()) # 打印按月份返回的json数据 MemoAdmin.memoadmin(data_file_path) # 利用数据文件名进行操作 config.config(data_file_path) # 用户配置文件生成 log.info('用户配置文件生成')
def __init__(self, memo_list): self.log = mylog.my_logger( base_path + '\\log\\' + '\ MemoAdmin_log.log', logging.DEBUG, 'pengtaolog') self.memo_list = memo_list