code update

api/api_app.py

@@ -9,18 +9,17 @@ import pythoncom
 import pywintypes
 import requests
 import win32com.client
+import webbrowser
 from flask import Flask, jsonify, request, current_app, send_from_directory, Response, stream_with_context
 from flask_cors import CORS
 from requests_toolbelt import MultipartEncoder
 from win10toast import ToastNotifier
 
-from config import file_type_map, TARGET_URL
+from config import file_type_map, TARGET_URL, SERVER_POINT
 from tools.check import is_file_open_in_wps
 from tools.file_manager import get_file_md5
 from tools.logger_handle import logger
 
-
-
 socket.getfqdn = lambda name=None: 'localhost'
 toaster = ToastNotifier()
 
@@ -28,6 +27,7 @@ app = Flask(__name__, static_folder='../static')
 
 CORS(app, resources=r'/*')
 
+
 class InterceptHandler(logging.Handler):
     def emit(self, record):
         # 将标准日志记录转为 loguru
@@ -426,6 +426,19 @@ def refresh_token():
     return jsonify(app.config['serve_client'].login_reply)
 
 
+@logger.catch()
+@app.route('/get_server_url', methods=['GET'])
+def get_server_url():
+    return jsonify({'code': 1000, 'msg': '操作成功', 'data': {'server_url': SERVER_POINT}})
+
+
+@logger.catch()
+@app.route('/open_platform', methods=['GET'])
+def open_platform():
+    webbrowser.open(SERVER_POINT)
+    return jsonify({'code': 1000, 'msg': '操作成功'})
+
+
 @logger.catch()
 @app.route('/download_cloud_file', methods=['POST'])
 def download_cloud_file():

api/ws_app.py

@@ -10,6 +10,7 @@ from tools.serve_client import server_client
 from tools.check import is_file_open_in_wps
 from tools.logger_handle import logger
 
+
 def ws_send(client, server, data):
     server.send_message(client, json.dumps(data))
 
@@ -28,7 +29,7 @@ def on_message(client, server, data):
     try:
         data = json.loads(data)
     except Exception:
-        return ws_send({'code':2003, 'msg':'参数错误'})
+        return ws_send({'code': 2003, 'msg': '参数错误'})
 
     if data.get('option') == 'download_and_open_file':
         file_id = data['params'].get('file_id')
@@ -79,7 +80,6 @@ def start_ws_server():
     ws_server = WebsocketServer(host='0.0.0.0', port=5856)
     ws_server.set_fn_message_received(on_message)
 
-    from tools.logger_handle import logger
     logger.info('ws server start')
     ws_server.run_forever()
 

config.py

@@ -5,8 +5,8 @@ import sys
 VERSION = '1.2.6'
 
 
-# dl_code = os.path.realpath(sys.argv[0])[:1]
-dl_code = 'd'
+dl_code = os.path.realpath(sys.argv[0])[:1]
+# dl_code = 'd'
 
 if not os.path.exists(f'{dl_code}:\\ProgramData\\OfficeAssistant\\config.json'):
     sys.exit(0)