136 lines
5.2 KiB
Python
136 lines
5.2 KiB
Python
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)
|
|
|
|
|
|
|