i}ndZddlmZddlmZ ddlZn #e$rdZYnwxYwddlmZGddZ dS)zSMySQL async client wrapper for persistent research session storage and convert log.) annotations)AnyN)settingscbeZdZdZddZeddZdd Zdd Zdd Z dddZ dddZ d S) MySQLClientz6Thin async wrapper around an aiomysql connection pool.poolrreturnNonec||_dSN)_pool)selfrs 9D:\work\zm-rag\backend\app\infrastructure\mysql_client.py__init__zMySQLClient.__init__s  'MySQLClient'c DKttdtjtjtjtjtjtjtj dtj tj tj  d{V}||S)Naiomysql is not installedT) hostportuserpassworddbcharset autocommitconnect_timeoutminsizemaxsize) aiomysql RuntimeError create_poolr mysql_host mysql_port mysql_usermysql_passwordmysql_database mysql_charsetmysql_connect_timeoutmysql_pool_minsizemysql_pool_maxsize)clsrs r from_settingszMySQLClient.from_settingss  :;; ;)$$$,&*$://          s4yyrczK|j|jd{VdSr )r close wait_closedrs rr.zMySQLClient.close'sF j$$&&&&&&&&&&&rc@K|dd{VdS)Na CREATE TABLE IF NOT EXISTS research_sessions ( session_id VARCHAR(128) NOT NULL, user_id VARCHAR(128) NOT NULL, history JSON NOT NULL, created_at DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP, updated_at DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP, expires_at DATETIME NULL, PRIMARY KEY (session_id, user_id), KEY idx_research_sessions_expires_at (expires_at), KEY idx_research_sessions_updated_at (updated_at) ) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_unicode_ci executer0s rensure_research_sessions_tablez*MySQLClient.ensure_research_sessions_table+sIll            rc@K|dd{VdS)Nu| CREATE TABLE IF NOT EXISTS convert_log ( id BIGINT UNSIGNED NOT NULL AUTO_INCREMENT, doc_id VARCHAR(256) NOT NULL COMMENT '文档 ID', task_id VARCHAR(128) DEFAULT NULL COMMENT 'Celery 任务 ID', source_format VARCHAR(32) NOT NULL COMMENT '源文件格式 (doc/xls/ppt/wps/et/ofd)', target_format VARCHAR(32) NOT NULL COMMENT '目标格式 (docx/xlsx/pptx/pdf)', source_path VARCHAR(1024) DEFAULT NULL COMMENT '源文件路径', output_path VARCHAR(1024) DEFAULT NULL COMMENT '转换后文件路径', success TINYINT(1) NOT NULL DEFAULT 0 COMMENT '是否成功 (0=失败 1=成功)', duration_ms INT DEFAULT NULL COMMENT '转换耗时(毫秒)', file_size BIGINT DEFAULT NULL COMMENT '源文件大小(字节)', error_message TEXT DEFAULT NULL COMMENT '失败时的错误信息', created_at DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP COMMENT '创建时间', PRIMARY KEY (id), KEY idx_convert_log_doc_id (doc_id), KEY idx_convert_log_task_id (task_id), KEY idx_convert_log_created_at (created_at) ) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_unicode_ci COMMENT='文档格式转换日志' r2r0s rensure_convert_log_tablez$MySQLClient.ensure_convert_log_table<sIll            rNquerystrparamstuple[Any, ...] | Nonedict[str, Any] | NonecKttd|j4d{V}|tj4d{V}|||pdd{V|d{V}dddd{Vn#1d{VswxYwYdddd{Vn#1d{VswxYwY|rt|ndS)Nr) rr r acquirecursor DictCursorr3fetchonedict)rr7r9connr?rows r fetch_onezMySQLClient.fetch_oneUs  :;; ;:%%'' . . . . . . .4{{8#677 . . . . . . .6nnUFLb999999999"OO-------- . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . )tCyyyT)s5&C 9B)C ) B3 3C 6B3 7C  CCintcfK|j4d{V}|4d{V}|||pdd{V}dddd{Vn#1d{VswxYwYdddd{Vn#1d{VswxYwYt |pdS)Nr=r)r r>r?r3rF)rr7r9rCr?affecteds rr3zMySQLClient.executebs :%%'' E E E E E E E4{{}} E E E E E E E!'v|!D!DDDDDDD E E E E E E E E E E E E E E E E E E E E E E E E E E E E E E E E E E E E E E E E E E E E E E E E E E E E E E8=q!!!s4BA.B. A8 8B;A8 <B BB)rrr r )r r)r r r )r7r8r9r:r r;)r7r8r9r:r rF) __name__ __module__ __qualname____doc__r classmethodr,r.r4r6rEr3r=rrrrs@@["''''    "    8*. * * * * * *."""""""rr) rL __future__rtypingrrModuleNotFoundError app.configrrr=rrrRsYY""""""OOOOHHH ["["["["["["["["["["s