# -*- coding: UTF-8 -*-
'''
@Project :domainScanDemo
@File    :remove_beian_time_field.py
@IDE     :PyCharm
@Author  :梦伴
@Date    :2026/4/11 22:40
@explain : 移除domains表中的beian_time字段
'''

import psycopg2
from loguru import logger
from app.config import config

def remove_beian_time_field():
    """
    移除domains表中的beian_time字段
    """
    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("数据库连接成功")
        
        # 移除beian_time字段
        try:
            cur.execute("ALTER TABLE domains DROP COLUMN IF EXISTS beian_time")
            logger.info("移除beian_time字段成功")
        except Exception as e:
            logger.error(f"移除beian_time字段失败: {e}")
        
        # 提交事务
        conn.commit()
        logger.info("字段移除完成")
        
        # 关闭连接
        cur.close()
        conn.close()
        
    except Exception as e:
        logger.error(f"执行失败: {e}")

if __name__ == "__main__":
    remove_beian_time_field()