debug

SecAegis · SecAegis · commit cb2784f95ca7 · 2025-06-15T22:20:22.000+08:00
diff --git a/device/block/bgp/bgp.py b/device/block/bgp/bgp.py
@@ -47,10 +47,10 @@ def check_exist_ip(ip) -> bool:
         server_port=80,
         sk="sk-*****",
         client_type="block",
-        block_ip = block_ip,
-        unblock_ip = unblock_ip,
-        get_all_block_ip= get_all_block_ip,
-        enable_cidr = True
+        block_ip=block_ip,
+        unblock_ip=unblock_ip,
+        get_all_block_ip=get_all_block_ip,
+        enable_cidr=True
     )
     sec_auto_ban.run()
     
diff --git a/device/block/check_point/check_point.py b/device/block/check_point/check_point.py
@@ -169,9 +169,9 @@ def check_exist_ip(ip) -> bool:
         server_port=80,
         sk="sk-*****",
         client_type="block",
-        block_ip = block_ip,
-        unblock_ip = unblock_ip,
-        get_all_block_ip= get_all_block_ip
+        block_ip=block_ip,
+        unblock_ip=unblock_ip,
+        get_all_block_ip=get_all_block_ip
     )
     pool = ThreadPool(processes=1)
     login()
diff --git a/device/block/dingtalk_robot/dingtalk_robot.py b/device/block/dingtalk_robot/dingtalk_robot.py
@@ -56,9 +56,9 @@ def check_exist_ip(ip) -> bool:
         server_port=80,
         sk="sk-*****",
         client_type="block",
-        block_ip = block_ip,
-        unblock_ip = unblock_ip,
-        enable_cidr = True
+        block_ip=block_ip,
+        unblock_ip=unblock_ip,
+        enable_cidr=True
     )
     pool = ThreadPool(processes=1)
     pool.apply_async(push)
diff --git a/device/block/example/base_example_fat.py b/device/block/example/base_example_fat.py
@@ -30,9 +30,9 @@ def check_exist_ip(ip) -> bool:
         server_port=80,
         sk="sk-*****",
         client_type="block",
-        block_ip = block_ip,
-        unblock_ip = unblock_ip,
-        get_all_block_ip= get_all_block_ip,
-        enable_cidr = False
+        block_ip=block_ip,
+        unblock_ip=unblock_ip,
+        get_all_block_ip=get_all_block_ip,
+        enable_cidr=False
     )
     sec_auto_ban.run()
diff --git a/device/block/example/base_example_thin.py b/device/block/example/base_example_thin.py
@@ -17,8 +17,8 @@ def unblock_ip(ip):
         server_port=80,
         sk="sk-*****",
         client_type="block",
-        block_ip = block_ip,
-        unblock_ip = unblock_ip,
-        enable_cidr = False
+        block_ip=block_ip,
+        unblock_ip=unblock_ip,
+        enable_cidr=False
     )
     sec_auto_ban.run()
diff --git a/device/block/opnsense/opnsense.py b/device/block/opnsense/opnsense.py
@@ -58,8 +58,8 @@ def check_exist_ip(ip) -> bool:
         server_port=80,
         sk="sk-*****",
         client_type="block",
-        block_ip = block_ip,
-        unblock_ip = unblock_ip,
-        get_all_block_ip= get_all_block_ip
+        block_ip=block_ip,
+        unblock_ip=unblock_ip,
+        get_all_block_ip=get_all_block_ip
     )
     sec_auto_ban.run()
diff --git a/device/block/qianxin_firewall/qianxin_firewall.py b/device/block/qianxin_firewall/qianxin_firewall.py
@@ -124,8 +124,8 @@ def check_exist_ip(cookies, ip) -> bool:
         server_port=80,
         sk="sk-*****",
         client_type="block",
-        block_ip = block_ip,
-        unblock_ip = unblock_ip,
-        get_all_block_ip= get_all_block_ip
+        block_ip=block_ip,
+        unblock_ip=unblock_ip,
+        get_all_block_ip=get_all_block_ip
     )
     sec_auto_ban.run()
diff --git a/device/block/router_os/router_os.py b/device/block/router_os/router_os.py
@@ -83,8 +83,8 @@ def check_exist_ip(ip) -> bool:
         server_port=80,
         sk="sk-*****",
         client_type="block",
-        block_ip = block_ip,
-        unblock_ip = unblock_ip,
-        get_all_block_ip= get_all_block_ip
+        block_ip=block_ip,
+        unblock_ip=unblock_ip,
+        get_all_block_ip=get_all_block_ip
     )
     sec_auto_ban.run()
diff --git a/device/block/tcp_reset/tcp_reset.py b/device/block/tcp_reset/tcp_reset.py
@@ -114,9 +114,9 @@ def run_sniff():
         server_port=80,
         sk="sk-*****",
         client_type="block",
-        block_ip = block_ip,
-        unblock_ip = unblock_ip,
-        get_all_block_ip= get_all_block_ip
+        block_ip=block_ip,
+        unblock_ip=unblock_ip,
+        get_all_block_ip=get_all_block_ip
     )
     pool = ThreadPool(processes=1)
     sec_auto_ban.print("[*] 开始拦截")
diff --git a/device/block/workweixin_robot/workweixin_robot.py b/device/block/workweixin_robot/workweixin_robot.py
@@ -1,6 +1,5 @@
 import time
 import requests
-import urllib.parse
 from SecAutoBan import SecAutoBan
 from multiprocessing.pool import ThreadPool
 
@@ -50,9 +49,9 @@ def check_exist_ip(ip) -> bool:
         server_port=80,
         sk="sk-*****",
         client_type="block",
-        block_ip = block_ip,
-        unblock_ip = unblock_ip,
-        enable_cidr = True
+        block_ip=block_ip,
+        unblock_ip=unblock_ip,
+        enable_cidr=True
     )
     pool = ThreadPool(processes=1)
     pool.apply_async(push)
diff --git a/device/sdk/python/README.md b/device/sdk/python/README.md
@@ -21,7 +21,7 @@ sec_auto_ban = SecAutoBan(
     server_port=8000,
     sk="sk-*****",
     client_type="alarm",
-    alarm_analysis = alarm_analysis
+    alarm_analysis=alarm_analysis
 )
 sec_auto_ban.run()
 ```
@@ -51,10 +51,10 @@ sec_auto_ban = SecAutoBan(
     server_port=8000,
     sk="sk-*****",
     client_type="block",
-    block_ip = block_ip,
-    unblock_ip = unblock_ip,
-    get_all_block_ip= get_all_block_ip,
-    enable_cidr = False
+    block_ip=block_ip,
+    unblock_ip=unblock_ip,
+    get_all_block_ip=get_all_block_ip,
+    enable_cidr=False
 )
 sec_auto_ban.run()
 ```
diff --git a/device/sdk/python/SecAutoBan/sec_auto_ban.py b/device/sdk/python/SecAutoBan/sec_auto_ban.py
@@ -5,14 +5,14 @@
 class SecAutoBan:
     init = False
     
-    def __init__(self, server_ip, server_port, sk, client_type, alarm_analysis=None, block_ip=None, unblock_ip=None, get_all_block_ip=None):
+    def __init__(self, server_ip, server_port, sk, client_type, enable_cidr=False, alarm_analysis=None, block_ip=None, unblock_ip=None, get_all_block_ip=None):
         self.client_type = client_type
         if client_type == "alarm":
             if alarm_analysis is None:
                 util.print("[-] 初始化失败: 未实现告警函数")
                 return
             self.alarm_analysis = alarm_analysis
-        self.ws_client = WebSocketClient(server_ip, server_port, sk, client_type, block_ip, unblock_ip, get_all_block_ip)
+        self.ws_client = WebSocketClient(server_ip, server_port, sk, client_type, enable_cidr, block_ip, unblock_ip, get_all_block_ip)
         self.init = True
 
     def print(self, message):
diff --git a/device/sdk/python/setup.py b/device/sdk/python/setup.py
@@ -5,7 +5,7 @@
 
 setuptools.setup(
     name="SecAutoBan",
-    version="3.2.0",
+    version="3.2.1",
     author="SecReport",
     author_email="secaegis@outlook.com",
     description="SecAutoBan SDK",
