diff --git a/roles/importer/files/importer/model_controllers/fwconfig_import_rule.py b/roles/importer/files/importer/model_controllers/fwconfig_import_rule.py index 244f3c601..6e93d0fb2 100644 --- a/roles/importer/files/importer/model_controllers/fwconfig_import_rule.py +++ b/roles/importer/files/importer/model_controllers/fwconfig_import_rule.py @@ -180,7 +180,7 @@ def collect_all_hit_information(self, prev_config: FwConfigNormalized, new_hit_i def add_hit_update(new_hit_information: list[dict], rule: RuleNormalized): """Add a hit information update entry for a rule.""" new_hit_information.append({ - "where": { "rule_uid": { "_eq": rule.rule_uid } }, + "where": { "rule_uid": { "_eq": rule.rule_uid }, "mgm_id": { "_eq": self.import_details.MgmDetails.CurrentMgmId } }, "_set": { "rule_last_hit": rule.last_hit } }) @@ -728,6 +728,7 @@ def PrepareNewRuleMetadata(self, newRules: list[Rulebase]) -> list[dict]: for rule_uid, rule in rulebase.rules.items(): rm4import = RuleMetadatum( rule_uid=rule_uid, + mgm_id=self.import_details.MgmDetails.CurrentMgmId, rule_last_modified=now, rule_created=now, rule_last_hit=rule.last_hit, diff --git a/roles/importer/files/importer/models/rule_metadatum.py b/roles/importer/files/importer/models/rule_metadatum.py index 6a0536430..e35575395 100644 --- a/roles/importer/files/importer/models/rule_metadatum.py +++ b/roles/importer/files/importer/models/rule_metadatum.py @@ -27,6 +27,7 @@ # Rule is the model for a normalized rule_metadata class RuleMetadatum(BaseModel): rule_uid: str + mgm_id: int rule_created: str|None = None rule_last_modified: str|None = None rule_first_hit: str|None = None