Upload db.py with huggingface_hub

db.py

@@ -248,6 +248,42 @@ def init_db():
         CREATE INDEX IF NOT EXISTS idx_reports_patient ON reports(patient_id);
         CREATE INDEX IF NOT EXISTS idx_reports_date ON reports(report_date);
         CREATE INDEX IF NOT EXISTS idx_ab_report ON antibody_strength(report_id);
+
+        CREATE TABLE IF NOT EXISTS dsa_reports (
+            id INTEGER PRIMARY KEY AUTOINCREMENT,
+            patient_id INTEGER NOT NULL,
+            report_date TEXT NOT NULL,
+            dsa_class TEXT NOT NULL,
+            pct_sa INTEGER NOT NULL,
+            overall TEXT NOT NULL,
+            specificity TEXT,
+            comment TEXT,
+            status TEXT DEFAULT 'draft',
+            submitted_by TEXT DEFAULT '',
+            upload_file TEXT DEFAULT '',
+            is_deleted INTEGER DEFAULT 0,
+            deleted_at TEXT,
+            created_at TEXT DEFAULT (datetime('now','localtime')),
+            updated_at TEXT DEFAULT (datetime('now','localtime')),
+            FOREIGN KEY (patient_id) REFERENCES patients(id)
+        );
+
+        CREATE TABLE IF NOT EXISTS dsa_antibody_strength (
+            id INTEGER PRIMARY KEY AUTOINCREMENT,
+            report_id INTEGER NOT NULL,
+            antigen TEXT NOT NULL,
+            allele TEXT,
+            strength TEXT,
+            max_mfi REAL,
+            mean_mfi REAL,
+            no_of_beads INTEGER,
+            bead_ids TEXT,
+            FOREIGN KEY (report_id) REFERENCES dsa_reports(id) ON DELETE CASCADE
+        );
+
+        CREATE INDEX IF NOT EXISTS idx_dsa_reports_patient ON dsa_reports(patient_id);
+        CREATE INDEX IF NOT EXISTS idx_dsa_reports_date ON dsa_reports(report_date);
+        CREATE INDEX IF NOT EXISTS idx_dsa_ab_report ON dsa_antibody_strength(report_id);
     ''')
     # migrate patients table
     pat_cols = [r[1] for r in conn.execute('PRAGMA table_info(patients)').fetchall()]
@@ -603,6 +639,267 @@ def get_mfi_comparison(chart_no, pra_class=None):
     return dates, antigens_out, pra_by_date, labels_by_rid
 
 
+# ============================================================
+# DSA helpers (Single-Antigen / Donor-Specific Antibody)
+# ============================================================
+
+def find_active_duplicate_dsa(patient_id, report_date, dsa_class):
+    conn = get_conn()
+    row = conn.execute(
+        '''SELECT id FROM dsa_reports
+           WHERE patient_id=? AND report_date=? AND dsa_class=?
+             AND COALESCE(is_deleted,0)=0''',
+        (patient_id, report_date, dsa_class)
+    ).fetchone()
+    conn.close()
+    return row['id'] if row else None
+
+
+def save_dsa_report(patient_id, report_date, dsa_class, pct_sa, overall,
+                    specificity, comment, sero_mfi_list, status='draft',
+                    submitted_by='', upload_file='', mode='overwrite'):
+    backup_db()
+    conn = get_conn()
+    existing = None
+    if mode == 'overwrite':
+        existing = conn.execute(
+            '''SELECT id FROM dsa_reports
+               WHERE patient_id=? AND report_date=? AND dsa_class=?
+                 AND COALESCE(is_deleted,0)=0''',
+            (patient_id, report_date, dsa_class)
+        ).fetchone()
+
+    if existing:
+        report_id = existing['id']
+        conn.execute('''UPDATE dsa_reports SET pct_sa=?, overall=?, specificity=?, comment=?,
+                        status=?, submitted_by=?, upload_file=?,
+                        updated_at=datetime('now','localtime') WHERE id=?''',
+                     (pct_sa, overall, specificity, comment, status, submitted_by,
+                      upload_file or '', report_id))
+        conn.execute('DELETE FROM dsa_antibody_strength WHERE report_id=?', (report_id,))
+    else:
+        cur = conn.execute(
+            '''INSERT INTO dsa_reports (patient_id, report_date, dsa_class, pct_sa, overall,
+               specificity, comment, status, submitted_by, upload_file)
+               VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)''',
+            (patient_id, report_date, dsa_class, pct_sa, overall, specificity, comment,
+             status, submitted_by, upload_file or ''))
+        report_id = cur.lastrowid
+
+    for m in sero_mfi_list:
+        conn.execute(
+            '''INSERT INTO dsa_antibody_strength
+               (report_id, antigen, allele, strength, max_mfi, mean_mfi, no_of_beads, bead_ids)
+               VALUES (?, ?, ?, ?, ?, ?, ?, ?)''',
+            (report_id, m.get('sero', ''), m.get('alleles', ''),
+             m.get('strength', ''), m.get('max_mfi', 0), m.get('mean_mfi', 0),
+             m.get('count', 0), m.get('beads', '')))
+
+    conn.commit()
+    conn.close()
+    schedule_auto_push()
+    return report_id
+
+
+def get_all_dsa_reports(limit=200):
+    conn = get_conn()
+    rows = conn.execute('''
+        SELECT r.*, p.patient_name, p.chart_no, p.donor_hla
+        FROM dsa_reports r JOIN patients p ON p.id = r.patient_id
+        WHERE COALESCE(r.is_deleted, 0) = 0
+        ORDER BY p.patient_name, p.chart_no, r.report_date DESC, r.dsa_class
+        LIMIT ?
+    ''', (limit,)).fetchall()
+    conn.close()
+    return [dict(r) for r in rows]
+
+
+def get_all_dsa_patients():
+    conn = get_conn()
+    rows = conn.execute('''
+        SELECT p.id, p.patient_name, p.chart_no,
+               COUNT(r.id) as report_count,
+               MAX(r.report_date) as latest_date,
+               GROUP_CONCAT(DISTINCT r.dsa_class) as classes
+        FROM patients p
+        LEFT JOIN dsa_reports r ON r.patient_id = p.id AND COALESCE(r.is_deleted, 0) = 0
+        GROUP BY p.id
+        HAVING report_count > 0
+        ORDER BY latest_date DESC
+    ''').fetchall()
+    conn.close()
+    return [dict(r) for r in rows]
+
+
+def get_dsa_patient_reports(chart_no):
+    conn = get_conn()
+    patient = conn.execute('SELECT * FROM patients WHERE chart_no=?', (chart_no,)).fetchone()
+    if not patient:
+        conn.close()
+        return None, []
+    reports = conn.execute('''
+        SELECT * FROM dsa_reports WHERE patient_id=? AND COALESCE(is_deleted,0)=0
+        ORDER BY report_date DESC, dsa_class
+    ''', (patient['id'],)).fetchall()
+    out = []
+    for r in reports:
+        ab = conn.execute('SELECT * FROM dsa_antibody_strength WHERE report_id=? ORDER BY id',
+                          (r['id'],)).fetchall()
+        out.append({**dict(r), 'antibodies': [dict(a) for a in ab]})
+    conn.close()
+    return dict(patient), out
+
+
+def get_dsa_mfi_comparison(chart_no, dsa_class=None):
+    """Same shape as PRA's get_mfi_comparison but reads dsa_reports."""
+    conn = get_conn()
+    patient = conn.execute('SELECT id FROM patients WHERE chart_no=?', (chart_no,)).fetchone()
+    if not patient:
+        conn.close()
+        return [], [], {}, {}
+
+    q = '''SELECT id, report_date, pct_sa, specificity FROM dsa_reports
+           WHERE patient_id = ? AND COALESCE(is_deleted, 0) = 0'''
+    params = [patient['id']]
+    if dsa_class:
+        q += ' AND dsa_class = ?'
+        params.append(dsa_class)
+    q += ' ORDER BY report_date, id'
+    reports = conn.execute(q, params).fetchall()
+
+    from collections import defaultdict
+    date_count = defaultdict(int)
+    for r in reports:
+        date_count[r['report_date']] += 1
+    date_seen = defaultdict(int)
+    labels_by_rid = {}
+    dates = []
+    pct_by_date = {}
+    for r in reports:
+        d = r['report_date']
+        if date_count[d] == 1:
+            label = d
+        else:
+            suffix = chr(ord('A') + date_seen[d])
+            label = f'{d} ({suffix})'
+            date_seen[d] += 1
+        labels_by_rid[r['id']] = label
+        dates.append(label)
+        pct_by_date[label] = r['pct_sa']
+
+    if not reports:
+        conn.close()
+        return [], [], {}, {}
+
+    rep_ids = [r['id'] for r in reports]
+    placeholders = ','.join('?' * len(rep_ids))
+    ab_rows = conn.execute(f'''
+        SELECT report_id, antigen, allele, max_mfi, mean_mfi, no_of_beads
+        FROM dsa_antibody_strength
+        WHERE report_id IN ({placeholders})
+        ORDER BY antigen
+    ''', rep_ids).fetchall()
+    conn.close()
+
+    import re as _re
+    parsed = {}
+    for r in reports:
+        spec = r['specificity'] or ''
+        sa_map = {}
+        bare = set()
+        for mt in _re.finditer(r'([A-Za-z][A-Za-z0-9]*)\(([^)]*)\)', spec):
+            s = mt.group(1).strip()
+            for a in mt.group(2).split():
+                a = a.strip()
+                if a:
+                    sa_map.setdefault(s, set()).add(a)
+        no_parens = _re.sub(r'\([^)]*\)', ' ', spec)
+        for tok in no_parens.split():
+            tok = tok.strip()
+            if '*' in tok and ':' in tok:
+                bare.add(tok)
+        parsed[r['id']] = {'sero_alleles': sa_map, 'bare': bare}
+
+    allele_to_sero = {}
+    for p in parsed.values():
+        for s, alleles in p['sero_alleles'].items():
+            for a in alleles:
+                allele_to_sero[a] = s
+
+    ab_map = {}
+    for ab in ab_rows:
+        ab_map[(ab['report_id'], ab['antigen'])] = dict(ab)
+
+    all_alleles = set()
+    for p in parsed.values():
+        for alleles in p['sero_alleles'].values():
+            all_alleles.update(alleles)
+        all_alleles.update(p['bare'])
+
+    def sk(a):
+        order = {'A': 0, 'B': 1, 'C': 2, 'Cw': 2,
+                 'DR': 3, 'DRB1': 3, 'DRB3': 3, 'DRB4': 3, 'DRB5': 3,
+                 'DQ': 4, 'DQB1': 4, 'DQA1': 5,
+                 'DP': 6, 'DPB1': 6, 'DPA1': 7}
+        loc = a.split('*')[0] if '*' in a else (_re.match(r'[A-Za-z]+', a) or _re.match(r'.', a)).group()
+        nums = _re.findall(r'\d+', a)
+        n1 = int(nums[0]) if nums else 0
+        n2 = int(nums[1]) if len(nums) > 1 else 0
+        return (order.get(loc, 99), n1, n2, a)
+
+    out = []
+    for allele in sorted(all_alleles, key=sk):
+        sero = allele_to_sero.get(allele)
+        mfi_by_date = {}
+        for r in reports:
+            rid = r['id']
+            lbl = labels_by_rid[rid]
+            p = parsed[rid]
+            present = allele in p['sero_alleles'].get(sero, set()) if sero else allele in p['bare']
+            if not present:
+                continue
+            ab = ab_map.get((rid, sero)) if sero else None
+            if not ab:
+                ab = ab_map.get((rid, allele))
+            if ab:
+                mfi_by_date[lbl] = {
+                    'max_mfi': ab['max_mfi'],
+                    'mean_mfi': ab['mean_mfi'],
+                    'no_of_beads': ab['no_of_beads'],
+                }
+        if not mfi_by_date:
+            continue
+        out.append({'antigen': sero if sero else allele, 'allele': allele,
+                    'mfi_by_date': mfi_by_date})
+    return dates, out, pct_by_date, labels_by_rid
+
+
+def delete_dsa_report(report_id):
+    backup_db()
+    conn = get_conn()
+    row = conn.execute('SELECT upload_file FROM dsa_reports WHERE id=?', (report_id,)).fetchone()
+    upload_file = row['upload_file'] if row and row['upload_file'] else ''
+    conn.execute("""UPDATE dsa_reports SET is_deleted=1, deleted_at=datetime('now','localtime'),
+                    updated_at=datetime('now','localtime') WHERE id=?""", (report_id,))
+    conn.commit()
+    if upload_file:
+        still_used = conn.execute(
+            'SELECT COUNT(*) FROM dsa_reports WHERE upload_file=? AND id!=? AND COALESCE(is_deleted,0)=0',
+            (upload_file, report_id)
+        ).fetchone()[0]
+        # Also check PRA reports
+        still_used += conn.execute(
+            'SELECT COUNT(*) FROM reports WHERE upload_file=? AND COALESCE(is_deleted,0)=0',
+            (upload_file,)
+        ).fetchone()[0]
+        conn.close()
+        if still_used == 0:
+            delete_upload(upload_file)
+    else:
+        conn.close()
+    schedule_auto_push()
+
+
 def delete_report(report_id):
     """軟刪除報告（標記 is_deleted=1，不真的刪）。
     若 upload_file 不再被任何 active report 引用，同步刪除 uploads/ 下的檔案。"""