|
@@ -41,12 +41,7 @@ def format_push_rules_for_user(
|
|
|
|
|
|
rulearray.append(template_rule)
|
|
|
|
|
|
- for type_key in ("pattern", "value"):
|
|
|
- type_value = template_rule.pop(f"{type_key}_type", None)
|
|
|
- if type_value == "user_id":
|
|
|
- template_rule[type_key] = user.to_string()
|
|
|
- elif type_value == "user_localpart":
|
|
|
- template_rule[type_key] = user.localpart
|
|
|
+ _convert_type_to_value(template_rule, user)
|
|
|
|
|
|
template_rule["enabled"] = enabled
|
|
|
|
|
@@ -63,19 +58,20 @@ def format_push_rules_for_user(
|
|
|
for c in template_rule["conditions"]:
|
|
|
c.pop("_cache_key", None)
|
|
|
|
|
|
- pattern_type = c.pop("pattern_type", None)
|
|
|
- if pattern_type == "user_id":
|
|
|
- c["pattern"] = user.to_string()
|
|
|
- elif pattern_type == "user_localpart":
|
|
|
- c["pattern"] = user.localpart
|
|
|
-
|
|
|
- sender_type = c.pop("sender_type", None)
|
|
|
- if sender_type == "user_id":
|
|
|
- c["sender"] = user.to_string()
|
|
|
+ _convert_type_to_value(c, user)
|
|
|
|
|
|
return rules
|
|
|
|
|
|
|
|
|
+def _convert_type_to_value(rule_or_cond: Dict[str, Any], user: UserID) -> None:
|
|
|
+ for type_key in ("pattern", "value"):
|
|
|
+ type_value = rule_or_cond.pop(f"{type_key}_type", None)
|
|
|
+ if type_value == "user_id":
|
|
|
+ rule_or_cond[type_key] = user.to_string()
|
|
|
+ elif type_value == "user_localpart":
|
|
|
+ rule_or_cond[type_key] = user.localpart
|
|
|
+
|
|
|
+
|
|
|
def _add_empty_priority_class_arrays(d: Dict[str, list]) -> Dict[str, list]:
|
|
|
for pc in PRIORITY_CLASS_MAP.keys():
|
|
|
d[pc] = []
|