import random import string from support.base.file import SupportFile from .setup import * name = 'setting' class ModuleSetting(PluginModuleBase): db_default = { 'db_version' : '1', 'port' : '9999', 'ddns' : 'http://localhost:9999', 'use_login' : 'False', 'web_id': 'admin', 'web_pw': 'Vm51JgZqhpwXc/UPc9CAN1lhj4s65+4ikv7GzNmvN6c=', 'web_title': 'Home', 'use_apikey': 'False', 'apikey': ''.join(random.choice(string.ascii_uppercase + string.digits) for _ in range(10)), f'restart_interval': f'{random.randint(0,59)} {random.randint(1,23)} * * *', 'theme' : 'Cerulean', 'log_level' : '20', 'plugin_dev_path': os.path.join(F.config['path_data'], 'dev'), 'system_start_time': '', # notify 'notify_telegram_use' : 'False', 'notify_telegram_token' : '', 'notify_telegram_chat_id' : '', 'notify_telegram_disable_notification' : 'False', 'notify_discord_use' : 'False', 'notify_discord_webhook' : '', 'notify_advaned_use' : 'False', } def __init__(self, P): super(ModuleSetting, self).__init__(P, name=name, first_menu='basic') def process_menu(self, page, req): arg = P.ModelSetting.to_dict() try: if page == 'config': arg['config.yaml'] = SupportFile.read_file(F.config['config_filepath']) arg['config_filepath'] = F.config['config_filepath'] elif page == 'export': arg['export_filepath'] = F.config['export_filepath'] if F.config['exist_export']: arg['export.sh'] = SupportFile.read_file(export) else: arg['export.sh'] = "export.sh 파일이 없습니다." elif page == 'menu': arg['menu_yaml_filepath'] = F.config['menu_yaml_filepath'] arg['menu.yaml'] = SupportFile.read_file(arg['menu_yaml_filepath']) return render_template(f'{__package__}_{name}_{page}.html', arg=arg) except Exception as e: P.logger.error(f'Exception:{str(e)}') P.logger.error(traceback.format_exc()) return render_template('sample.html', title=f"{__package__}/{name}/{page}") def process_command(self, command, arg1, arg2, arg3, req): ret = {'ret':'success'} if command == 'apikey_generate': return jsonify(''.join(random.choice(string.ascii_uppercase + string.digits) for _ in range(10))) elif command == 'config_save': SupportFile.write_file(F.config['config_filepath'], arg1 ) ret['msg'] = '저장하였습니다.' elif command == 'export_save': if F.config['exist_export']: SupportFile.write_file(F.config['export_filepath'], arg1 ) ret['msg'] = '저장하였습니다.' else: ret['ret'] = 'warning' ret['msg'] = 'export.sh 파일이 없습니다.' elif command == 'menu_save': SupportFile.write_file(F.config['menu_yaml_filepath'], arg1 ) ret['msg'] = '저장하였습니다.' from framework.init_menu import MenuManager MenuManager.init_menu() F.socketio.emit("refresh", {}, namespace='/framework', broadcast=True) elif command == 'notify_test': if arg1 == 'telegram': pass return jsonify(ret) def plugin_load(self): try: if F.config['arg_repeat'] == 0 or SystemModelSetting.get('system_start_time') == '': SystemModelSetting.set('system_start_time', datetime.now().strftime('%Y-%m-%d %H:%M:%S')) SystemModelSetting.set('repeat', str(F.config['arg_repeat'])) username = SystemModelSetting.get('web_id') passwd = SystemModelSetting.get('web_pw') F.users[username] = User(username, passwd_hash=passwd) self.__set_restart_scheduler() self.__set_scheduler_check_scheduler() F.get_recent_version() except Exception as e: P.logger.error(f'Exception:{str(e)}') P.logger.error(traceback.format_exc()) def setting_save_after(self, change_list): if 'theme' in change_list: F.socketio.emit("refresh", {}, namespace='/framework', broadcast=True) def __set_restart_scheduler(self): name = f'{__package__}_restart' if F.scheduler.is_include(name): F.scheduler.remove_job(name) interval = SystemModelSetting.get('restart_interval') if interval != '0': if len(interval.split(' ')) == 1: interval = '%s' % (int(interval) * 60) job_instance = Job(__package__, name, interval, F.restart, "자동 재시작") F.scheduler.add_job_instance(job_instance, run=False) def __set_scheduler_check_scheduler(self): name = 'scheduler_check' if F.scheduler.is_include(name): F.scheduler.remove_job(name) job_instance = Job(__package__, name, 2, F.scheduler.first_run_check_thread_function, "Scheduler Check") scheduler.add_job_instance(job_instance, run=False)