From 28def91c7d04016c5fc5b21da51a5fc2aad01b1f Mon Sep 17 00:00:00 2001
From: Mr Chen <midoks@163.com>
Date: Tue, 12 Nov 2024 22:56:40 +0800
Subject: [PATCH] Update __init__.py

---
 web/admin/__init__.py | 51 +++++++++++++++++++++----------------------
 1 file changed, 25 insertions(+), 26 deletions(-)

diff --git a/web/admin/__init__.py b/web/admin/__init__.py
index 776c95b9d..a33c569ff 100644
--- a/web/admin/__init__.py
+++ b/web/admin/__init__.py
@@ -66,32 +66,6 @@ app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = True
 # 初始化db
 setup.init()
 
-# webssh
-socketio = SocketIO()
-socketio.init_app(app)
-
-@socketio.on('webssh_websocketio')
-def webssh_websocketio(data):
-    if not isLogined():
-        emit('server_response', {'data': '会话丢失，请重新登陆面板!\r\n'})
-        return
-    import utils.ssh.ssh_terminal as ssh_terminal
-    shell_client = ssh_terminal.ssh_terminal.instance()
-    shell_client.run(request.sid, data)
-    return
-
-
-@socketio.on('webssh')
-def webssh(data):
-    if not isLogined():
-        emit('server_response', {'data': '会话丢失，请重新登陆面板!\r\n'})
-        return None
-
-    import utils.ssh.ssh_local as ssh_local
-    shell = ssh_local.ssh_local.instance()
-    shell.run(data)
-    return
-
 # BASIC AUTH
 app.config['BASIC_AUTH_OPEN'] = False
 try:
@@ -171,6 +145,31 @@ def inject_global_variables():
     }
     return dict(config=g_config, data=data)
 
+# webssh
+socketio = SocketIO()
+socketio.init_app(app)
+
+@socketio.on('webssh_websocketio')
+def webssh_websocketio(data):
+    if not isLogined():
+        emit('server_response', {'data': '会话丢失，请重新登陆面板!\r\n'})
+        return
+    import utils.ssh.ssh_terminal as ssh_terminal
+    shell_client = ssh_terminal.ssh_terminal.instance()
+    shell_client.run(request.sid, data)
+    return
+
+
+@socketio.on('webssh')
+def webssh(data):
+    if not isLogined():
+        emit('server_response', {'data': '会话丢失，请重新登陆面板!\r\n'})
+        return None
+
+    import utils.ssh.ssh_local as ssh_local
+    shell = ssh_local.ssh_local.instance()
+    shell.run(data)
+    return
 
 # Log the startup
 app.logger.info('########################################################')