Ver Fonte

feat: 优化代码

cheng zhen há 1 ano atrás
pai
commit
ae4ade33e2
3 ficheiros alterados com 17 adições e 69 exclusões
  1. 1 1
      browser_utils.py
  2. 15 67
      license_manager.py
  3. 1 1
      server/.env

+ 1 - 1
browser_utils.py

@@ -29,7 +29,7 @@ class BrowserManager:
         co.set_pref("credentials_enable_service", False)
         co.set_argument("--hide-crash-restore-bubble")
         co.auto_port()
-        co.headless(True)  # 生产环境使用无头模式
+        # co.headless(True)  # 生产环境使用无头模式
 
         # Mac 系统特殊处理
         if sys.platform == "darwin":

+ 15 - 67
license_manager.py

@@ -84,8 +84,6 @@ class LicenseManager:
         """在线激活许可证"""
         try:
             machine_code = self.get_hardware_info()
-
-            # 准备激活数据
             activation_data = {
                 "license_key": license_key,
                 "machine_code": machine_code,
@@ -93,20 +91,13 @@ class LicenseManager:
             }
 
             try:
-                print("\n正在连接激活服务器...")
-                print("发送激活请求数据:", activation_data)
-
                 response = requests.post(
                     self.activation_url, json=activation_data, timeout=10
                 )
 
-                print(f"服务器响应状态码: {response.status_code}")
-                print(f"服务器响应内容: {response.text}")
-
                 if response.status_code == 200:
                     result = response.json()
                     if result.get("success"):
-                        # 保存许可证信息
                         license_data = {
                             "license_key": license_key,
                             "machine_code": machine_code,
@@ -118,29 +109,17 @@ class LicenseManager:
                         return True, "激活成功!"
                     else:
                         error_msg = result.get("message", "激活失败")
-                        print(f"激活失败: {error_msg}")
                         return False, error_msg
                 else:
-                    error_msg = f"服务器响应错误: HTTP {response.status_code}"
-                    print(error_msg)
-                    return False, error_msg
+                    return False, f"服务器响应错误: HTTP {response.status_code}"
 
-            except requests.exceptions.ConnectionError as e:
-                error_msg = f"无法连接到激活服务器: {str(e)}"
-                print(error_msg)
+            except requests.exceptions.ConnectionError:
                 return False, "无法连接到激活服务器,请检查网络连接"
             except requests.exceptions.Timeout:
-                error_msg = "服务器响应超时"
-                print(error_msg)
-                return False, error_msg
+                return False, "服务器响应超时"
 
         except Exception as e:
-            error_msg = f"激活过程出错: {str(e)}"
-            print(error_msg)
-            import traceback
-
-            print(traceback.format_exc())
-            return False, error_msg
+            return False, f"激活过程出错: {str(e)}"
 
     def verify_license(self):
         """验证许可证"""
@@ -152,71 +131,41 @@ class LicenseManager:
             if not license_data:
                 return False, "许可证文件无效"
 
-            # 验证机器码
             current_machine = self.get_hardware_info()
             if current_machine != license_data.get("machine_code"):
                 return False, "硬件信息不匹配"
 
-            # 在线验证
             verify_data = {
                 "license_key": license_data.get("license_key"),
                 "machine_code": current_machine,
             }
 
             try:
-                print("正在验证许可证...")
-                print("发送验证请求数据:", verify_data)
-
-                # 设置更长的超时时间
-                response = requests.post(
-                    self.verify_url, json=verify_data, timeout=30  # 增加超时时间到30秒
-                )
-
-                # 等待响应
-                time.sleep(2)  # 添加短暂延迟
+                response = requests.post(self.verify_url, json=verify_data, timeout=30)
 
-                print(f"服务器响应状态码: {response.status_code}")
+                time.sleep(2)
 
                 try:
-                    response_text = response.text
-                    print(f"服务器响应内容: {response_text}")
                     result = response.json()
-                except Exception as e:
-                    print(f"解析响应失败: {str(e)}")
+                except Exception:
                     return False, "服务器响应格式错误"
 
                 if response.status_code == 200:
                     if result.get("success"):
-                        print("许可证验证成功")
                         return True, "许可证有效"
-                    error_msg = result.get("message", "许可证无效")
-                    print(f"验证失败: {error_msg}")
-                    return False, error_msg
+                    return False, result.get("message", "许可证无效")
 
-                error_msg = f"服务器响应错误: HTTP {response.status_code}"
-                print(error_msg)
-                return False, error_msg
+                return False, f"服务器响应错误: HTTP {response.status_code}"
 
-            except requests.exceptions.ConnectionError as e:
-                error_msg = f"无法连接到验证服务器: {str(e)}"
-                print(error_msg)
+            except requests.exceptions.ConnectionError:
                 return False, "无法连接到验证服务器,请检查网络连接"
             except requests.exceptions.Timeout:
-                error_msg = "服务器响应超时"
-                print(error_msg)
-                return False, error_msg
+                return False, "服务器响应超时"
             except Exception as e:
-                error_msg = f"验证请求失败: {str(e)}"
-                print(error_msg)
-                return False, error_msg
+                return False, f"验证请求失败: {str(e)}"
 
         except Exception as e:
-            error_msg = f"验证过程出错: {str(e)}"
-            print(error_msg)
-            import traceback
-
-            print(traceback.format_exc())
-            return False, error_msg
+            return False, f"验证过程出错: {str(e)}"
 
     def _save_license(self, license_data):
         """加密保存许可证数据"""
@@ -226,7 +175,7 @@ class LicenseManager:
             with open(self.license_file, "w") as f:
                 f.write(encrypted_data)
         except Exception as e:
-            print(f"保存许可证出错: {e}")
+            pass
 
     def _load_license(self):
         """加密读取许可证数据"""
@@ -235,6 +184,5 @@ class LicenseManager:
                 encrypted_data = f.read()
             decrypted_data = self.decrypt(encrypted_data)
             return json.loads(decrypted_data)
-        except Exception as e:
-            print(f"读取许可证出错: {e}")
+        except Exception:
             return None

+ 1 - 1
server/.env

@@ -1,4 +1,4 @@
-MONGODB_URI=mongodb://mongo_3y6JyM:mongo_iNySJ5@119.8.35.41:27017/license-manager?authSource=admin&retryWrites=true&w=majority
+MONGODB_URI=mongodb://mongo_3y6JyM:mongo_iNySJ5@119.8.35.41:27017/license-manager?authSource=admin&retryWrites=true&w=majority&tz=Asia/Shanghai
 PORT=3000
 ENCRYPTION_KEY=f1e2d3c4b5a6978899aabbccddeeff00112233445566778899aabbccddeeff00
 MAX_USAGE_COUNT=10