idZddlZddlZddlZddlZej j ej jeZ ej je dZ ejZ daejZgdZdZdZdZd Zd Zd Zed k(reed e edeyy)u SQLite storage backend — drop-in replacement for JSON file I/O. Exposes load_json_file / save_json_file with identical signatures to the previous file-based helpers, so request handlers need no change. - WAL mode: concurrent reads while a writer is active - timeout=30s: writers queue instead of failing under contention - ACID: no partial writes; crash-safe - Auto-migrate: first run imports existing *.json files from BASE_DIR Nz database.dbF) z products.jsonzcustomers.jsonzcert_history.jsonzdictionary.jsonzvisual_template_configs.jsonzcert_config.jsonztemplate_fields.jsonzfastener_specs.jsonz packing.jsonzfield_mapping.jsonctjtdd}|jd|jd|jd|S)N)timeoutisolation_levelzPRAGMA journal_mode=WALzPRAGMA synchronous=NORMALzPRAGMA foreign_keys=ON)sqlite3connectDB_PATHexecute)conns /app/cer_product_mecsu/db.py _get_connr 's@ ??7B EDLL*+LL,-LL)* KcRtryt5tr dddyt} |jd|jdj }|ddk(r t ||j dadddy#|j wxYw#1swYyxYw)Nz CREATE TABLE IF NOT EXISTS kv_store ( key TEXT PRIMARY KEY, value TEXT NOT NULL, updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP ) zSELECT COUNT(*) FROM kv_storerT) _init_done _init_lockr r fetchone _auto_migrateclose)r rows r _ensure_initr/s   {  LL ,,>?HHJC1v{d# JJL # JJL!s(B BAB-BBBB&cDg}tD]}tjjt|}tjj |sG t |dd5}tj|}ddd|dd}tjdd}|jd ||f|j||r*td t|d dj|yy#1swYzxYw#t$r}td |d |Yd}~d}~wwxYw)z8Import existing *.json files into kv_store on first run.rzutf-8)encodingNF ensure_asciiindentz:INSERT OR REPLACE INTO kv_store (key, value) VALUES (?, ?)z [migrate] skip z: z [migrate] imported z files into SQLite: z, )MIGRATABLE_FILESospathjoinBASE_DIRexistsopenjsonloaddumpsr append Exceptionprintlen) r importedfnr!fdatakeyvaluees r rrGs H1ww||Hb)ww~~d#  1dC'2 $ayy| $Sb'CJJt%BE LLLe   OOB 1  %c(m_4HS[I\H]^_ $ $ 1 %bTA3/ 0 0 1s1C:C.5A C:.C7 3C:: DDDc t|jdr|ddn|}t} |jd|fj }|j |rt j|dSgS#|j wxYw)zHDrop-in replacement. Returns [] if key not found (matches old behavior)..jsonNrz(SELECT value FROM kv_store WHERE key = ?r)rendswithr r rrr&loads)filenamer1r rs r load_json_filer9^szN#,,W5(3B-8C ;DllEvNWWY  zz#a&!! I s !A;;B c8t|jdr|ddn|}tj|dd}t5t } |j d||f|j dddy#|jwxYw#1swYyxYw)z4Drop-in replacement. Atomic write via SQLite UPSERT.r5NrFrrzINSERT INTO kv_store (key, value, updated_at) VALUES (?, ?, CURRENT_TIMESTAMP) ON CONFLICT(key) DO UPDATE SET value = excluded.value, updated_at = CURRENT_TIMESTAMP)rr6r&r( _write_lockr r r)r8r0r1r2r s r save_json_filer<lsN#,,W5(3B-8C JJt% :E  {  LLhe   JJL   JJL  s$ B A;!B;B  BBctt} |jdDcgc]}|d c}|jScc}w#|jwxYw)u.Debug helper — returns all keys in kv_store.z%SELECT key FROM kv_store ORDER BY keyr)rr r r)r rs r list_keysr>}sIN ;D"ll+RST!T U sA AA A A__main__zDB:zKeys:)__doc__r&r r threadingr!dirnameabspath__file__r#r"r Lockr;rrrr rrr9r<r>__name__r+rr rHs   77??277??84 5 '',,x /inn  Y^^   0`. " zN % '9;r