v2.0.1 — 修复 database.py 缺少聚合函数

This commit is contained in:
mac
2026-07-24 16:23:07 +08:00
parent 5b637e7fed
commit 5b73fc0eae

View File

@@ -224,3 +224,90 @@ def reorder_wishes(order_list):
conn.execute('UPDATE wishes SET sort_order = ? WHERE id = ?', (idx, wid))
conn.commit()
conn.close()
def get_weeks_list():
"""返回所有已有打卡记录的周号列表"""
conn = get_conn()
rows = conn.execute('SELECT DISTINCT date FROM checkins ORDER BY date').fetchall()
conn.close()
return [r['date'] for r in rows]
def get_daily_backup(week_str):
"""获取某周的原始日记录(从备份表)"""
from datetime import datetime as _dt, timedelta
if '-W' not in week_str:
return []
year_part, week_part = week_str.split('-W')
try:
year = int(year_part)
week = int(week_part)
except (ValueError, IndexError):
return []
jan1 = _dt(year, 1, 1)
delta = timedelta(days=(week - 1) * 7 - jan1.weekday())
monday = jan1 + delta
dates = [monday.strftime('%Y-%m-%d')]
for i in range(1, 7):
dates.append((monday + timedelta(days=i)).strftime('%Y-%m-%d'))
conn = get_conn()
placeholders = ','.join('?' for _ in dates)
exists = conn.execute("SELECT name FROM sqlite_master WHERE type='table' AND name='checkins_daily_backup'").fetchone()
if not exists:
conn.close()
return []
rows = conn.execute(
f"SELECT date, data FROM checkins_daily_backup WHERE date IN ({placeholders}) ORDER BY date",
dates
).fetchall()
conn.close()
results = []
for row in rows:
results.append({'date': row['date'], 'data': json.loads(row['data'])})
return results
def _aggregate_daily_to_weekly(conn):
"""将 checkins_daily_backup 的日数据聚合为周数据,写入 checkins"""
from datetime import datetime as _dt
rows = conn.execute('SELECT date, data FROM checkins_daily_backup ORDER BY date').fetchall()
weekly = {}
for row in rows:
try:
d = _dt.strptime(row['date'], '%Y-%m-%d')
except ValueError:
continue
iso_year, iso_week, _ = d.isocalendar()
week_key = f"{iso_year}-W{iso_week:02d}"
if week_key not in weekly:
weekly[week_key] = {'morning': [], 'evening': [], 'study': []}
data = json.loads(row['data'])
for mi in data.get('morning', []):
text = mi if isinstance(mi, str) else mi.get('text', '')
if text:
existing_texts = {x if isinstance(x, str) else x.get('text', '') for x in weekly[week_key]['morning']}
if text not in existing_texts:
weekly[week_key]['morning'].append(mi)
for ei in data.get('evening', []):
mst = ei.get('mistake', '') if isinstance(ei, dict) else ''
if mst:
existing = {x.get('mistake', '') if isinstance(x, dict) else x for x in weekly[week_key]['evening']}
if mst not in existing:
weekly[week_key]['evening'].append(ei if isinstance(ei, dict) else {'mistake': ei, 'improvement': ''})
existing_names = {}
for si in weekly[week_key]['study']:
nm = si.get('name', '') if isinstance(si, dict) else str(si)
if nm:
existing_names[nm] = si
for si in data.get('study', []):
nm = si.get('name', '') if isinstance(si, dict) else str(si)
if nm and nm not in existing_names:
weekly[week_key]['study'].append(si)
existing_names[nm] = si
now = _dt.now().isoformat()
for week_key in sorted(weekly.keys()):
conn.execute(
'INSERT INTO checkins (date, data, created_at, updated_at) VALUES (?, ?, ?, ?)',
(week_key, json.dumps(weekly[week_key], ensure_ascii=False), now, now)
)