Upload db.py with huggingface_hub

db.py

@@ -109,6 +109,11 @@ def init_db():
 
     # migrate: soft delete columns
     report_cols_all = [r[1] for r in conn.execute('PRAGMA table_info(reports)').fetchall()]
+    if 'upload_file' not in report_cols_all:
+        try:
+            conn.execute("ALTER TABLE reports ADD COLUMN upload_file TEXT DEFAULT ''")
+        except Exception:
+            pass
     if 'is_deleted' not in report_cols_all:
         try:
             conn.execute("ALTER TABLE reports ADD COLUMN is_deleted INTEGER DEFAULT 0")
@@ -179,7 +184,8 @@ def get_donor_hla(chart_no):
 
 
 def save_report(patient_id, report_date, pra_class, pra_percent, overall,
-                specificity, comment, sero_mfi_list, status='draft', submitted_by=''):
+                specificity, comment, sero_mfi_list, status='draft', submitted_by='',
+                upload_file=''):
     backup_db()
     conn = get_conn()
     existing = conn.execute(
@@ -190,14 +196,14 @@ def save_report(patient_id, report_date, pra_class, pra_percent, overall,
     if existing:
         report_id = existing['id']
         conn.execute('''UPDATE reports SET pra_percent=?, overall=?, specificity=?, comment=?,
-                        status=?, submitted_by=?, updated_at=datetime('now','localtime') WHERE id=?''',
-                     (pra_percent, overall, specificity, comment, status, submitted_by, report_id))
+                        status=?, submitted_by=?, upload_file=?, updated_at=datetime('now','localtime') WHERE id=?''',
+                     (pra_percent, overall, specificity, comment, status, submitted_by, upload_file or '', report_id))
         conn.execute('DELETE FROM antibody_strength WHERE report_id=?', (report_id,))
     else:
         cur = conn.execute(
             '''INSERT INTO reports (patient_id, report_date, pra_class, pra_percent, overall,
-               specificity, comment, status, submitted_by) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)''',
-            (patient_id, report_date, pra_class, pra_percent, overall, specificity, comment, status, submitted_by))
+               specificity, comment, status, submitted_by, upload_file) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)''',
+            (patient_id, report_date, pra_class, pra_percent, overall, specificity, comment, status, submitted_by, upload_file or ''))
         report_id = cur.lastrowid
 
     for m in sero_mfi_list:
@@ -348,6 +354,22 @@ def delete_report(report_id):
     conn.close()
 
 
+UPLOAD_DIR = DB_PATH.parent / 'uploads'
+
+
+def get_upload_dir():
+    UPLOAD_DIR.mkdir(exist_ok=True)
+    return UPLOAD_DIR
+
+
+def get_storage_stats():
+    """取得 uploads 資料夾使用量"""
+    d = get_upload_dir()
+    files = list(d.glob('*.xls')) + list(d.glob('*.xlsx')) + list(d.glob('*.csv'))
+    total_size = sum(f.stat().st_size for f in files)
+    return {'count': len(files), 'size_mb': round(total_size / 1024 / 1024, 2)}
+
+
 def register_user(username, password, display_name=''):
     """註冊新使用者"""
     backup_db()