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