diff --git a/plugins/ollama/index.py b/plugins/ollama/index.py index 05eff64d6..6885c8b4d 100755 --- a/plugins/ollama/index.py +++ b/plugins/ollama/index.py @@ -29,8 +29,6 @@ class App: def __init__(self): self.__setupPath = self.getServerDir() - self.__cfg = self.__setupPath + '/nezha.cfg' - self.__agent_cfg = self.__setupPath + '/agent.cfg' def getArgs(self): args = sys.argv[3:] @@ -54,22 +52,9 @@ class App: return (False, mw.returnJson(False, '参数:(' + ck[i] + ')没有!')) return (True, mw.returnJson(True, 'ok')) - def __release_port(self, port): - from collections import namedtuple - try: - import firewall_api - firewall_api.firewall_api().addAcceptPortArgs(port, 'nezha', 'port') - return port - except Exception as e: - return "Release failed {}".format(e) - - def openPort(self): - for i in ["9527", "5555"]: - self.__release_port(i) - return True def getPluginName(self): - return 'nezha' + return 'ollama' def getPluginDir(self): return mw.getPluginDir() + '/' + self.getPluginName() @@ -77,14 +62,6 @@ class App: def getServerDir(self): return mw.getServerDir() + '/' + self.getPluginName() - def getInitdConfTpl(self): - path = self.getPluginDir() + "/init.d/nezha.tpl" - return path - - def getInitdAgentConfTpl(self): - path = self.getPluginDir() + "/init.d/nezha-agent.tpl" - return path - def getHomeDir(self): if mw.isAppleSystem(): user = mw.execShell( @@ -102,20 +79,12 @@ class App: return 'root' def status(self): - cmd = "ps -ef|grep " + self.getPluginName() + \ - " |grep -v grep | grep -v nezha-agent | grep -v python | awk '{print $2}'" + cmd = "ps -ef|grep " + self.getPluginName() + " |grep -v grep | grep -v python | awk '{print $2}'" data = mw.execShell(cmd) if data[0] == '': return "stop" return 'start' - def status_agent(self): - cmd = "ps -ef | grep nezha-agent | grep -v grep | grep -v python | awk '{print $2}'" - data = mw.execShell(cmd) - if data[0] == '': - return 'stop' - return 'start' - def contentReplace(self, content): service_path = mw.getServerDir() @@ -145,15 +114,15 @@ class App: mw.execShell('chmod +x ' + file_bin) # systemd - systemDir = mw.systemdCfgDir() - systemService = systemDir + '/nezha.service' - systemServiceTpl = self.getPluginDir() + '/init.d/nezha.service.tpl' - if os.path.exists(systemDir) and not os.path.exists(systemService): - service_path = mw.getServerDir() - se_content = mw.readFile(systemServiceTpl) - se_content = self.contentReplace(se_content) - mw.writeFile(systemService, se_content) - mw.execShell('systemctl daemon-reload') + # systemDir = mw.systemdCfgDir() + # systemService = systemDir + '/nezha.service' + # systemServiceTpl = self.getPluginDir() + '/init.d/nezha.service.tpl' + # if os.path.exists(systemDir) and not os.path.exists(systemService): + # service_path = mw.getServerDir() + # se_content = mw.readFile(systemServiceTpl) + # se_content = self.contentReplace(se_content) + # mw.writeFile(systemService, se_content) + # mw.execShell('systemctl daemon-reload') return file_bin @@ -166,39 +135,11 @@ class App: return content - def initDAgent(self): - file_tpl = self.getInitdAgentConfTpl() - - initD_path = self.getServerDir() + '/init.d' - if not os.path.exists(initD_path): - os.mkdir(initD_path) - - file_agent_bin = initD_path + '/nezha-agent' - - content = mw.readFile(file_tpl) - content = self.contentReplace(content) - content = self.contentAgentReplace(content) - mw.writeFile(file_agent_bin, content) - mw.execShell('chmod +x ' + file_agent_bin) - - # systemd - sysDir = mw.systemdCfgDir() - sysService = sysDir + '/nezha-agent.service' - sysServiceTpl = self.getPluginDir() + '/init.d/nezha-agent.service.tpl' - service_path = mw.getServerDir() - content = mw.readFile(sysServiceTpl) - content = self.contentReplace(content) - content = self.contentAgentReplace(content) - mw.writeFile(sysService, content) - mw.execShell('systemctl daemon-reload') - - return file_agent_bin def init_cfg(self): self.initDreplace() - self.initDAgent() - def imOp(self, method): + def oaOp(self, method): file = self.initDreplace() @@ -215,26 +156,21 @@ class App: return data[0] def start(self): - return self.imOp('start') + return self.oaOp('start') def stop(self): - return self.imOp('stop') + return self.oaOp('stop') def restart(self): - return self.imOp('restart') + return self.oaOp('restart') def reload(self): - return self.imOp('reload') + return self.oaOp('reload') def agOp(self, method): - file = self.initDAgent() - - path = self.__agent_cfg - if not os.path.exists(path): - return '请先设置Agent配置!' if not mw.isAppleSystem(): - cmd = 'systemctl {} {}'.format(method, 'nezha-agent') + cmd = 'systemctl {} {}'.format(method, self.getPluginName()) data = mw.execShell(cmd) if data[1] == '': return 'ok' @@ -258,101 +194,23 @@ class App: return self.agentOp('reload') def initd_status(self): - cmd = 'systemctl status nezha | grep loaded | grep "enabled;"' + cmd = 'systemctl status '+self.getPluginName()+' | grep loaded | grep "enabled;"' data = mw.execShell(cmd) if data[0] == '': return 'fail' return 'ok' def initd_install(self): - mw.execShell('systemctl enable nezha') + mw.execShell('systemctl enable '+self.getPluginName()) return 'ok' def initd_uninstall(self): - mw.execShell('systemctl disable nezha') - return 'ok' - - def initd_status_agent(self): - cmd = 'systemctl status nezha-agent | grep loaded | grep "enabled;"' - data = mw.execShell(cmd) - if data[0] == '': - return 'fail' - return 'ok' - - def initd_install_agent(self): - mw.execShell('systemctl enable nezha-agent') - return 'ok' - - def initd_uninstall_agent(self): - mw.execShell('systemctl disable nezha-agent') + mw.execShell('systemctl disable '+self.getPluginName()) return 'ok' def conf(self): return self.getServerDir() + '/dashboard/data/config.yaml' - def nezha_cfg(self): - path = self.__cfg - if not os.path.exists(path): - d = {} - cmd_un = 'cd ' + self.getServerDir() + '/dashboard && ./nezha conf -u ""' - mw.execShell(cmd_un) - td = mw.execShell(cmd_un) - d['username'] = td[0].strip() - - pwd = mw.getRandomString(16) - cmd_pwd = 'cd ' + self.getServerDir() + '/dashboard && ./nezha conf -u "" -p ' + pwd - td = mw.execShell(cmd_pwd) - d['password'] = pwd - - mw.writeFile(path, mw.enDoubleCrypt('nezha', mw.getJson(d))) - - info = mw.readFile(path) - info = mw.deDoubleCrypt('nezha', info) - - info = json.loads(info) - return mw.returnJson(True, 'ok', info) - - def nezha_save_cfg(self): - args = self.getArgs() - data = self.checkArgs(args, ['username', 'password']) - if not data[0]: - return data[1] - path = self.__cfg - - cmd = 'cd ' + self.getServerDir() + '/dashboard && ./nezha conf -su "' + \ - args['username'] + '" -p ' + args['password'] - # print(cmd) - t = mw.execShell(cmd) - # print(t) - mw.writeFile(path, mw.enDoubleCrypt('nezha', mw.getJson(args))) - return mw.returnJson(True, '修改成功!') - - def get_agent_cfg(self): - path = self.__agent_cfg - info = mw.readFile(path) - info = mw.deDoubleCrypt('agent', info) - info = json.loads(info) - return info - - def agent_cfg(self): - path = self.__agent_cfg - if not os.path.exists(path): - d = {} - d['host'] = '127.0.0.1:5555' - d['secret'] = 'secret' - mw.writeFile(path, mw.enDoubleCrypt('agent', mw.getJson(d))) - - info = self.get_agent_cfg() - return mw.returnJson(True, 'ok', info) - - def agent_save_cfg(self): - args = self.getArgs() - data = self.checkArgs(args, ['host', 'secret']) - if not data[0]: - return data[1] - path = self.__agent_cfg - mw.writeFile(path, mw.enDoubleCrypt('agent', mw.getJson(args))) - return mw.returnJson(True, '修改成功!') def run_log(self): return self.getServerDir() + '/logs/nezha.log'