69 lines
2.3 KiB
Python
69 lines
2.3 KiB
Python
"""补充更新 special_types 和 number_features（基于标题重算）"""
|
||
import re
|
||
import yaml
|
||
from datetime import datetime
|
||
|
||
def load_config():
|
||
with open('/root/coolbot-data/config.yaml', 'r') as f:
|
||
return yaml.safe_load(f)
|
||
|
||
config = load_config()
|
||
DB_CONFIG = config['database']
|
||
|
||
def get_db_conn():
|
||
import psycopg2
|
||
return psycopg2.connect(
|
||
host=DB_CONFIG['host'], port=DB_CONFIG.get('port', 5432),
|
||
user=DB_CONFIG['user'], password=DB_CONFIG['password'],
|
||
database=DB_CONFIG['database']
|
||
)
|
||
|
||
FEATURE_MAP = {
|
||
'倒置': ['倒置'], '如意': ['如意'], '朦胧': ['朦胧'], '金马': ['金马'], '金山': ['金山'],
|
||
'天马': ['天马'], '钻石': ['钻石'], '永恒': ['永恒'], '无4': ['无4', '无四'],
|
||
'无47': ['无47'], '无247': ['无247'], '无34': ['无34'], '无347': ['无347'],
|
||
'带4': ['带4'], '豹子': ['豹子'], '狮子': ['狮子'], '老虎': ['老虎'],
|
||
'大象': ['大象'], '生日': ['生日'], '满号': ['满号'], '首日': ['首日'],
|
||
}
|
||
|
||
SPECIAL_MAP = {
|
||
'标十': ['标十'], '标百': ['标百'],
|
||
'刀': ['刀', '刀货'], '单张': ['单张', '散张'],
|
||
'捆': ['捆'], '千连': ['千连'], '百连': ['百连'],
|
||
'救生圈': ['大救生圈', '救生圈'],
|
||
}
|
||
|
||
def calc_special(title):
|
||
if not title: return None
|
||
found = [k for k, words in SPECIAL_MAP.items() if any(w in title for w in words)]
|
||
return '|'.join(found) if found else None
|
||
|
||
def calc_features(title):
|
||
if not title: return None
|
||
found = [k for k, words in FEATURE_MAP.items() if any(w in title for w in words)]
|
||
return '|'.join(found) if found else None
|
||
|
||
conn = get_db_conn()
|
||
cur = conn.cursor()
|
||
cur.execute("SELECT id, title FROM yichens_posts WHERE special_types IS NULL OR number_features IS NULL")
|
||
rows = cur.fetchall()
|
||
print(f'需要更新: {len(rows)} 条')
|
||
|
||
updated = 0
|
||
for rid, title in rows:
|
||
sp = calc_special(title)
|
||
ft = calc_features(title)
|
||
cur.execute("""
|
||
UPDATE yichens_posts SET special_types = COALESCE(%s, special_types),
|
||
number_features = COALESCE(%s, number_features)
|
||
WHERE id = %s
|
||
""", (sp, ft, rid))
|
||
updated += 1
|
||
if updated % 200 == 0:
|
||
print(f'已更新 {updated}/{len(rows)}')
|
||
|
||
conn.commit()
|
||
cur.close()
|
||
conn.close()
|
||
print(f'完成! 共更新 {updated} 条')
|