h`)SSKrSSKrSSKrSSKrSSKJrJrJrJr SSK J r \R"\RSS9 \R"\5r"SS5rg)N)ListDictAnyOptional)get_db_managerz4%(asctime)s - %(name)s - %(levelname)s - %(message)s)levelformatc\rSrSrSrSS\S\4SjjrSrSrS\ \ \\ 44S jr S \S\ \ \\ 44S jrS \ \\ 4S\4S jrS \S\ \\ 4S\4SjrS \S\4SjrS \S\4SjrS \S\4SjrS\S\S\ \\ 4S\4SjrS\S\ \ \\ 44SjrS\ \ \\ 44SjrS\S\4SjrS\S\ \ \\ 4S\4SjrS\S\ \ \\ 44SjrSrg) RuleManager u> 规则管理器，负责规则的增删改查和持久化。 db_path rules_filecX l/Ul[5UlUS:waSSKJn U"U5UlUR 5 g)Ndata/honeypot.dbr)DatabaseManager)rrulesr db_manager core.databaser _load_rules)selfr rrs core/rule_manager.py__init__RuleManager.__init__s<$+- (* ( ( 5-g6DO c[RRUR5(a[ URSSS9n[ R "U5n[U[5(aURS/5O/Ul SSS5 [RS[UR5SUR35 g[R!S URS 35 /Ul g!,(df  Nr=f![a6n[RSURS U35 /Ul SnAgSnAff=f) u从文件加载全局规则rutf-8encodingrNzLoaded z global rules from zFailed to load rules from : z Rules file z/ not found. Initializing with empty rules list.)ospathexistsropenjsonload isinstancedictgetrloggerinfolen Exceptionerrorwarning)rfdataes rrRuleManager._load_ruless 77>>$// * * $//3AQ99Q== L LL3DOO3DBqcJ K K Ls7AC AB8/C 8 CC C D%C==Dreturnc6URR5$)u获取所有全局规则)rcopy)rs r get_all_rulesRuleManager.get_all_rules7szz  rrule_idc\URHnURS5U:XdMUs $ g)u根据ID获取全局规则idN)rr)rrDrules rget_rule_by_idRuleManager.get_rule_by_id;s*JJDxx~( rrHcURS5(d[RS5 gURUS5(a[RSUSS35 gURR U5 UR 5 g)u添加新全局规则rFzRule must have an 'id'FRule with ID 'z' already existsT)r)r*r.rIrappendr=)rrHs radd_ruleRuleManager.add_ruleBsrxx~~ LL1 2   tDz * * LL>$t*5EF G $ r updated_rulec[UR5H:up4URS5U:XdMX RU'UR5 g [R SUS35 g)u更新全局规则rFTrLz' not found for updateF enumeraterr)r=r*r/)rrDrPirHs r update_ruleRuleManager.update_ruleOs\ ,GAxx~( , 1   " - y0FGHrc[UR5H9up#URS5U:XdMURU UR5 g [R SUS35 g)u删除全局规则rFTrLz' not found for deletionFrR)rrDrTrHs r delete_ruleRuleManager.delete_ruleYs\ ,GAxx~(JJqM  " - y0HIJrc`URU5nU(aSUS'UR5 gg)u启用全局规则TenabledFrIr=rGs r enable_ruleRuleManager.enable_rulecs0""7+ "DO    rc`URU5nU(aSUS'UR5 gg)u禁用全局规则Fr[Tr\rGs r disable_ruleRuleManager.disable_rulels0""7+ #DO    renv_iddriverconfigc URR5nUR5nURSX[R "USS945 UR 5 [RSUS35 SSS5 g!,(df  g=f![Ra [RSUS35 g[Ra%n[RS US U35 SnAgSnAff=f) u 在数据库中创建一个新环境。 :param env_id: 环境唯一标识符。 :param driver: 使用的驱动名称。 :param config: 环境配置字典。 :return: 操作是否成功。 z>INSERT INTO environments (id, driver, config) VALUES (?, ?, ?)Fr8 Environment z created in database.NTz already exists in database.zError creating environment r )rget_connectioncursorexecuter%dumpscommitr*r+sqlite3IntegrityErrorr/Errorr.)rrbrcrdconnrir2s rcreate_environmentRuleManager.create_environmentws //1TTTZZU%KL  l6(2GHI211%%  NN\&1MN O}}  LL6vhbD E s;BA"B>B BBB-C?C?C::C?cURR5nUR5nURSU45 UR 5nU(a2[ U5n[ R"US5US'UsSSS5 $SSS5 g!,(df  g=f![Ra%n[RSUSU35 SnAgSnAff=f)u 从数据库获取环境信息。 :param env_id: 环境唯一标识符。 :return: 环境信息字典，如果未找到则返回 None。 z'SELECT * FROM environments WHERE id = ?rdNzError getting environment r ) rrhrirjfetchoner(r%loadsrmror*r.)rrbrprirowenv_datar2s rget_environmentRuleManager.get_environments E//1TH6)Too'#CyH)-HX4F)GHX&#21 221}} E LL5fXRsC D D Es;B$A#B? B$ B$ B!B$!B$$C8CCcURR5nUR5nURS5 UR 5n/nUH;n[ U5n[ R"US5US'URU5 M= UsSSS5 $!,(df  g=f![Ra$n[RSU35 /sSnA$SnAff=f)uH 从数据库列出所有环境。 :return: 环境信息字典列表。 zSELECT * FROM environmentsrdNzError listing environments: ) rrhrirjfetchallr(r%rurMrmror*r.)rrprirows environmentsrvrwr2s rlist_environmentsRuleManager.list_environmentss  //1T;<(! C#CyH)-HX4F)GHX& ''1  $211}}  LL7s; <I s;B-A6B B- B*&B-*B--C%C C% C%cURR5nUR5nURSU45 UR 5 UR S:a#[ RSUS35 SSS5 g[ RSUS35 SSS5 g!,(df  g=f![Ra%n[ RS US U35 SnAgSnAff=f) u 从数据库删除环境。由于设置了 CASCADE，environment_rules 表中的相关记录也会被自动删除。 :param env_id: 环境唯一标识符。 :return: 操作是否成功。 z%DELETE FROM environments WHERE id = ?rrgz deleted from database.NTz not found for deletion.FzError deleting environment r ) rrhrirjrlrowcountr*r+r/rmror.)rrbrprir2s rdelete_environmentRuleManager.delete_environments //1TF R ??Q&KK,vh6M NO 21NN\&9Q#RS 211}}  LL6vhbD E sAB7AB&:B7B&B7& B40B74B77C0 C++C0rcURR5nUR5nURSU45 [ U5H.upV[ R "USS9nURSXU45 M0 UR5 [RS[U5SUS35 SSS5 g !,(df  g=f![Ra%n[RS US U35 SnAgSnAff=f) u 为指定环境设置（覆盖）规则列表。 :param env_id: 环境唯一标识符。 :param rules: 要关联到环境的规则对象列表。 :return: 操作是否成功。 z6DELETE FROM environment_rules WHERE environment_id = ?FrfzVINSERT INTO environment_rules (environment_id, rule_data, rule_order) VALUES (?, ?, ?)zSet z rules for environment .NTz$Error setting rules for environment r )rrhrirjrSr%rkrlr*r+r,rmror.) rrbrrpriindexrH rule_jsonr2s rset_environment_rules!RuleManager.set_environment_ruless //1TWZ`Ybc$-U#3KE $ 4e DINNpE2$4  d3u:,.EfXQOP211 }}  LL?xr!M N s5C BB<3C < C C C D!DDcURR5nUR5nURSU45 UR 5n/nUH-n[ R "US5nURU5 M/ UsSSS5 $!,(df  g=f![Ra'n[RSUSU35 /sSnA$SnAff=f)u 获取指定环境的规则列表。 :param env_id: 环境唯一标识符。 :return: 与该环境关联的规则对象列表。 zTSELECT rule_data FROM environment_rules WHERE environment_id = ? ORDER BY rule_order rule_dataNz$Error getting rules for environment r ) rrhrirjr{r%rurMrmror*r.) rrbrprir|rrvrule_objr2s rget_environment_rules!RuleManager.get_environment_ruless  //1TjI(C#zz#k*:;HLL* 211}}  LL?xr!M NI  s;B!A*B B! BB!B!!C5CCC)rrrN)rzconfig/rules.json)__name__ __module__ __qualname____firstlineno____doc__strrrr=rrrrBrrIboolrNrUrXr]r`rqrxr~rrr__static_attributes__rrr r s  c  L!tDcN3!chtCH~.F T#s(^  3d38n3434CDc4S>VZ2chtCH~.F(4S#X#7**CT#s(^8LQU8CDc3h4Hrr )r%r!loggingrmtypingrrrrrr basicConfigINFO getLoggerrr*r rrrrsM ,,(',,/ef   8 $vvr