29 lines
786 B
Python
29 lines
786 B
Python
# -*- coding: utf-8 -*-
|
|
"""添加 custom_content 和 search_name 字段"""
|
|
import sqlite3
|
|
|
|
DB_PATH = r"D:\夏骥\微信研究\contacts.db"
|
|
|
|
conn = sqlite3.connect(DB_PATH)
|
|
cursor = conn.cursor()
|
|
|
|
# 检查字段是否已存在
|
|
cursor.execute("PRAGMA table_info(contacts)")
|
|
columns = [col[1] for col in cursor.fetchall()]
|
|
|
|
if 'custom_content' not in columns:
|
|
cursor.execute('ALTER TABLE contacts ADD COLUMN custom_content TEXT DEFAULT ""')
|
|
print("已添加 custom_content 字段")
|
|
else:
|
|
print("custom_content 字段已存在")
|
|
|
|
if 'search_name' not in columns:
|
|
cursor.execute('ALTER TABLE contacts ADD COLUMN search_name TEXT DEFAULT ""')
|
|
print("已添加 search_name 字段")
|
|
else:
|
|
print("search_name 字段已存在")
|
|
|
|
conn.commit()
|
|
conn.close()
|
|
print("完成")
|