恢复 detectPort

moeshin · Apr 2, 2023 · c69524f · c69524f
1 parent bf499aa
commit c69524f
Showing 1 changed file with 10 additions and 5 deletions.
diff --git a/utils/port.py b/utils/port.py
@@ -1,10 +1,15 @@
-import requests
+import socket
 
 
 def detectPort(port=6666):
-    if not port:
-        return False
+
+    s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
     try:
-        return requests.head("http://127.0.0.1:%d/ocr/api" % port, timeout=2).headers.get("Dango-OCR") == "OK"
+        s.connect(("127.0.0.1", int(port)))
+        s.shutdown(2)
+        sign = True
     except Exception:
-        return False
+        sign = False
+    s.close()
+
+    return sign