convert domainCheck to regular directory

2026-04-16 13:33:06 +08:00
parent e406d73334
commit d37c444929
107 changed files with 254185 additions and 1 deletions
--- a/domainCheck/add_fields_to_domains.py
+++ b/domainCheck/add_fields_to_domains.py
@@ -0,0 +1,59 @@
+# -*- coding: UTF-8 -*-
+'''
+@Project :domainScanDemo
+@File    :add_fields_to_domains.py
+@IDE     :PyCharm
+@Author  :梦伴
+@Date    :2026/4/11 22:40
+@explain : 为domains表添加缺失的字段
+'''
+
+import psycopg2
+from loguru import logger
+from app.config import config
+
+def add_fields_to_domains():
+    """
+    为domains表添加缺失的字段
+    """
+    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()
+        logger.info("数据库连接成功")
+        
+        # 添加company_type字段
+        try:
+            cur.execute("ALTER TABLE domains ADD COLUMN IF NOT EXISTS company_type VARCHAR(100)")
+            cur.execute("COMMENT ON COLUMN domains.company_type IS '单位性质'")
+            logger.info("添加company_type字段成功")
+        except Exception as e:
+            logger.error(f"添加company_type字段失败: {e}")
+        
+        # 添加website_url字段
+        try:
+            cur.execute("ALTER TABLE domains ADD COLUMN IF NOT EXISTS website_url VARCHAR(255)")
+            cur.execute("COMMENT ON COLUMN domains.website_url IS '网站首页网址'")
+            logger.info("添加website_url字段成功")
+        except Exception as e:
+            logger.error(f"添加website_url字段失败: {e}")
+        
+        # 提交事务
+        conn.commit()
+        logger.info("字段添加完成")
+        
+        # 关闭连接
+        cur.close()
+        conn.close()
+        
+    except Exception as e:
+        logger.error(f"执行失败: {e}")
+
+if __name__ == "__main__":
+    add_fields_to_domains()