convert domainCheck to regular directory
This commit is contained in:
50
domainCheck/remove_beian_time_field.py
Normal file
50
domainCheck/remove_beian_time_field.py
Normal file
@@ -0,0 +1,50 @@
|
||||
# -*- 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()
|
||||
Reference in New Issue
Block a user