convert domainCheck to regular directory

domainCheck/check_database.py

@@ -0,0 +1,74 @@
+# -*- coding: UTF-8 -*-
+'''
+@Project :domainScanDemo
+@File    :check_database.py
+@IDE     :PyCharm
+@Author  :梦伴
+@Date    :2026/4/9 21:40
+@explain : 检查数据库表结构和注释
+'''
+
+import psycopg2
+from app.config import config
+
+
+def check_database():
+    """
+    检查数据库表结构和注释
+    """
+    try:
+        # 连接数据库
+        conn = psycopg2.connect(
+            host=config.DB_HOST,
+            port=config.DB_PORT,
+            database=config.DB_DATABASE,
+            user=config.DB_USER,
+            password=config.DB_PASSWORD
+        )
+        cur = conn.cursor()
+        print("数据库连接成功")
+        
+        # 检查所有表
+        print("\n=== 所有表 ===")
+        cur.execute("SELECT table_name FROM information_schema.tables WHERE table_schema = 'public'")
+        tables = cur.fetchall()
+        for table in tables:
+            print(f"表名: {table[0]}")
+        
+        # 检查domains表结构
+        print("\n=== domains表结构 ===")
+        cur.execute("\d+ domains.txt")
+        result = cur.fetchall()
+        for row in result:
+            print(row)
+        
+        # 检查其他表结构
+        print("\n=== detect_tasks表结构 ===")
+        cur.execute("\d+ detect_tasks")
+        result = cur.fetchall()
+        for row in result:
+            print(row)
+        
+        print("\n=== domain_blacklist表结构 ===")
+        cur.execute("\d+ domain_blacklist")
+        result = cur.fetchall()
+        for row in result:
+            print(row)
+        
+        print("\n=== domain_detections表结构 ===")
+        cur.execute("\d+ domain_detections")
+        result = cur.fetchall()
+        for row in result:
+            print(row)
+        
+    except Exception as e:
+        print(f"检查数据库失败: {e}")
+    finally:
+        if cur:
+            cur.close()
+        if conn:
+            conn.close()
+
+
+if __name__ == "__main__":
+    check_database()