o h @sXddlZddlZddlZddlZejejddeeZGdddZ da ddZ dS)Nz4%(asctime)s - %(name)s - %(levelname)s - %(message)s)levelformatc@s*eZdZdZd ddZddZddZd S) DatabaseManageru6 管理 SQLite 数据库连接和初始化。 data/honeypot.dbcCs*||_tjtj|jdd|dS)uh 初始化数据库管理器。 :param db_path: SQLite 数据库文件路径。 T)exist_okN)db_pathosmakedirspathdirnameinit_db)selfrr/root/ive/core/database.py__init__s zDatabaseManager.__init__cCst|j}tj|_|S)uR 获取数据库连接。 :return: SQLite 连接对象。 )sqlite3connectrRow row_factory)r connrrrget_connections zDatabaseManager.get_connectionc Csz9|*}|}|d|d|d|td|jWdWdS1s2wYWdStjyO}z t d|d}~ww)u< 初始化数据库，创建必要的表。 a+ CREATE TABLE IF NOT EXISTS environments ( id TEXT PRIMARY KEY, driver TEXT NOT NULL, config TEXT NOT NULL, -- JSON string status TEXT NOT NULL ); u CREATE TABLE IF NOT EXISTS environment_rules ( environment_id TEXT NOT NULL, rule_data TEXT NOT NULL, -- JSON string of the rule object rule_order INTEGER NOT NULL, -- 用于保持规则顺序 FOREIGN KEY (environment_id) REFERENCES environments (id) ON DELETE CASCADE ); zUCREATE INDEX IF NOT EXISTS idx_env_rules_env_id ON environment_rules(environment_id);z%Database initialized successfully at NzError initializing database: ) rcursorexecutecommitloggerinforrErrorerror)r rrerrrr #s   &zDatabaseManager.init_dbN)r)__name__ __module__ __qualname____doc__rrr rrrrr s   rcCstdurtatS)uX 获取全局数据库管理器实例。 :return: DatabaseManager 实例。 N) db_managerrrrrrget_db_managerNsr$) rrjsonlogging basicConfigINFO getLoggerrrrr#r$rrrrs B