59 lines
1.7 KiB
Python
59 lines
1.7 KiB
Python
# -*- 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() |