spacefreak/pyquarantine-milter
1
0
Fork 0
Code Issues Pull Requests Releases Activity

add list condition

Browse Source
This commit is contained in:
Thomas Oettli
2023-12-12 14:13:08 +01:00
parent f42860d900
commit 479c1513a3
7 changed files with 52 additions and 66 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
pyquarantine/config.py
View File

@@ -22,7 +22,7 @@ __all__ = [
"RewriteLinksConfig",
"StoreConfig",
"NotifyConfig",
"AllowListConfig",
"ListConfig",
"QuarantineConfig",
"ActionConfig",
"RuleConfig",
@@ -89,7 +89,7 @@ class BaseConfig:
return self._config
class AllowListConfig(BaseConfig):
class ListConfig(BaseConfig):
JSON_SCHEMA = {
"type": "object",
"required": ["type"],
@@ -121,14 +121,14 @@ class ConditionsConfig(BaseConfig):
"headers": {"type": "array",
"items": {"type": "string"}},
"var": {"type": "string"},
"allowlist": {"type": "object"}}}
"list": {"type": "object"}}}
def __init__(self, config, rec=True):
super().__init__(config)
if not rec:
return
if "allowlist" in self:
self["allowlist"] = AllowListConfig(self["allowlist"])
if "list" in self:
self["list"] = ListConfig(self["list"])
class AddHeaderConfig(BaseConfig):
@@ -257,8 +257,7 @@ class QuarantineConfig(BaseConfig):
if "notify" in self:
self["notify"] = NotifyConfig(self["notify"])
if "allowlist" in self:
self["allowlist"] = ConditionsConfig(
{"allowlist": self["allowlist"]}, rec)
self["allowlist"] = ListConfig(self["allowlist"])
class ActionConfig(BaseConfig):