def sieve_is(self, content):
filtersset = FiltersSet("Testfilterset")
filtersset.from_parser_result(self.parser)
target = six.StringIO()
filtersset.tosieve(target)
repr_ = target.getvalue()
target.close()
self.assertEqual(repr_, content)
def test_get_filter_conditions_from_parser_result(self):
res = """require ["fileinto"];
# rule:[test]
if anyof (exists ["Subject"]) {
fileinto "INBOX";
}
"""
p = parser.Parser()
p.parse(res)
fs = FiltersSet("test", '# rule:')
fs.from_parser_result(p)
c = fs.get_filter_conditions('[test]')
self.assertEqual(c, [("exists", "Subject")])
res = """require ["date", "fileinto"];
# rule:aaa
if anyof (currentdate :zone "+0100" :is "date" ["2019-03-27"]) {
fileinto "INBOX";
}
"""
p = parser.Parser()
p.parse(res)
fs = FiltersSet("aaa", "# rule:")
fs.from_parser_result(p)
c = fs.get_filter_conditions('aaa')
self.assertEqual(
c,
[('currentdate', ':zone', '+0100', ':is', 'date', '2019-03-27')])
res = """require ["envelope", "fileinto"];
# rule:[aaa]
if anyof (envelope :contains ["To"] ["*****@*****.**"]) {
fileinto "INBOX";
}
"""
p = parser.Parser()
p.parse(res)
fs = FiltersSet("aaa", "# rule:")
fs.from_parser_result(p)
c = fs.get_filter_conditions('[aaa]')
self.assertEqual(
c, [('envelope', ':contains', ['To'], ['*****@*****.**'])])
def constructFilterSet(actions):
moreSieve = ""
print(actions)
fs = FiltersSet("test")
sieveContent = io.StringIO()
for action in actions:
print("cfS-> act ", action)
print("type---", type(action))
if type(action) == sievelib.commands.IfCommand:
#print("cfS-> act0 type tosieve", action.tosieve())
#fs.addfilter(action.name, action['test'], action.children)
action.tosieve(target=sieveContent)
#print(sieveContent.read())
moreSieve = moreSieve + "# Filter:\n" + sieveContent.getvalue()
#print("moreSieve", moreSieve)
else:
conditions = action[0][2]
#print("cfSS-> cond", conditions)
cond = []
#print("cond....", cond)
#print(conditions)
for condition in conditions:
#print("cfS condition -> ", condition)
# print(type(condition))
#print(condition.arguments)
head = ()
(key1, key2, key3) = list(condition.arguments.keys())
#print("keys",key1, key2, key3)
#print("keys",condition.arguments[key1], condition.arguments[key2], condition.arguments[key3])
head = head + (condition.arguments['header-names'].strip('"'),
condition.arguments['match-type'].strip('"'),
condition.arguments['key-list'].strip('"')
) #.decode('utf-8'))
# We will need to take care of these .decode's
#print("head", head)
cond.append(head)
#print("cond", cond)
# print "cond ->", cond
act = []
for i in range(len(action[0][1])):
act.append((action[0][0], action[0][1][i]))
act.append(("stop", ))
#print("cfS cond ->", cond)
#print("cfS act ->", act)
aList = [()]
#for a in cond[0]:
# print("cfS ",a)
# aList[0] = aList[0] + (a.strip('"'),)
#print("cfS aList", aList)
#print("cfS cond", cond)
#print("cfS act", act)
fs.addfilter("", cond, act)
#fs.addfilter("", aList, act)
# print "added!"
return (fs, moreSieve)
def getscript(self, name, format="raw"):
content = self.msc.getscript(name)
if content is None:
raise SieveClientError(self.msc.errmsg)
if format == "raw":
return content
p = Parser()
if not p.parse(content):
print "Parse error????"
return None
fs = FiltersSet(name)
fs.from_parser_result(p)
return fs
def constructFilterSet(actions):
fs = FiltersSet("test")
for action in actions:
#print("cfS-> act ", action)
conditions = action[0][2]
#print("cfS-> cond", conditions)
cond = []
for condition in conditions:
#print("cfS condition -> ", condition)
# print(type(condition))
#print(condition.arguments)
head = ()
(key1, key2, key3) = list(condition.arguments.keys())
#print("keys",key1, key2, key3)
#print("keys",condition.arguments[key1], condition.arguments[key2], condition.arguments[key3])
head = head + (condition.arguments['match-type'].strip('"'),
condition.arguments['header-names'].strip('"'),
condition.arguments['key-list'].strip('"')
) #.decode('utf-8'))
# We will need to take care of these .decode's
#print(head)
cond.append(head)
# print "cond ->", cond
act = []
for i in range(len(action[0][1])):
act.append((action[0][0], action[0][1][i]))
act.append(("stop", ))
#print("cfS cond ->", cond)
#print("cfS act ->", act)
aList = [()]
#for a in cond[0]:
# print("cfS ",a)
# aList[0] = aList[0] + (a.strip('"'),)
#print("cfS aList", aList)
#print("cfS cond", cond)
#print("cfS act", act)
fs.addfilter("", cond, act)
#fs.addfilter("", aList, act)
# print "added!"
return fs
def setUp(self):
self.fs = FiltersSet("test")
def get_filter_vars(data):
filter_match_types = {
"match_any_flowing_rules": "anyof",
"match_all_flowing_rules": "allof",
"match_all": "all",
}
filter_sections = {
"subject": "subject",
"from": "from",
"to": "to",
"cc": "cc",
"to_or_cc": "to_or_cc",
"size": "size",
"header": "header",
"body": "body"
}
filter_conditions = {
"is": ":is",
"contains": ":contains",
"matches": ":matches",
"matches_regex": ":regex",
"is_under": ":over",
"is_over": ":under",
}
filter_negative_conditions = {
"is_not": ":is",
"does_not_contain": ":contains",
"does_not_match": ":matches",
"does_not_match_regex": ":regex"
}
filter_section_conditions = {
"subject": lambda x: [x],
"from": lambda x: [x],
"to": lambda x: [x],
"cc": lambda x: [x],
"to_or_cc": lambda x: [x],
"size": lambda x: [x],
"header": lambda x: [x],
"body": lambda x: [":text", x]
}
filter_actions = {
"discard_message": "discard",
"keep_message": "keep",
"stop_processing_filter": "stop",
"forward_message_to": "redirect",
"send_reject_message": "reject",
"file_message_in": "fileinto"
}
action_value_map = {
"forward_message_to": "to_email",
"send_reject_message": "message",
"file_message_in": "folder"
}
filters = data.value
raw_filter = ""
fs = FiltersSet("cow")
filters = sorted(filters, key=lambda f: f['order'])
for s_filter in filters: # range(len(filters.keys())):
# filter= filters[i]
filter_name = s_filter.get("name")
enabled = True # s_filter.get("enabled") TODO ::
rules = s_filter.get("conditions")
actions = s_filter.get("actions")
matchtype = filter_match_types[s_filter.get("incoming_message")]
builded_conditions = []
for rule in rules: # range(len(rules.keys())):
# rule = rules[rule_number]
negated = False
section = filter_sections[rule["selector"]]
if "not" in rule["condition"]:
negated = True
condition = filter_negative_conditions[rule["condition"]]
else:
condition = filter_conditions[rule["condition"]]
condition = filter_section_conditions[section](condition)
if negated:
section = "not{}".format(section)
value = rule["value"]
builded_conditions.append((section, *condition, value))
builded_actions = []
actions = sorted(actions, key=lambda f: f['order'])
for action in actions: # range(len(actions.keys())):
# action = actions[action_number]
param = ""
if action["type"] in action_value_map.keys():
param = action[action_value_map[action["type"]]]
# if value_map=[]
builded_actions.append((filter_actions[action["type"]], param))
if matchtype == "all":
p = Parser()
raw = 'require ["reject","fileinto","imap4flags"];'
for builded_action in builded_actions:
name = builded_action[0]
param = builded_action[1]
if param:
raw += "\n{} \"{}\";".format(name, param)
else:
raw += "\n{};".format(name)
p.parse(raw)
fs.from_parser_result(p)
else:
fs.addfilter(name=filter_name,
conditions=builded_conditions,
actions=builded_actions,
matchtype=matchtype)
requirements = fs.requires
io = StringIO()
fs.tosieve(io)
raw_sieve = io.getvalue()
if requirements:
raw_sieve = '\n'.join(raw_sieve.splitlines()[2:])
if "not" in raw_sieve:
for key in filter_sections.keys():
raw_sieve = raw_sieve.replace("not" + key, key)
return [raw_sieve, requirements]
