-
Notifications
You must be signed in to change notification settings - Fork 2
/
add.py
executable file
·78 lines (60 loc) · 2.3 KB
/
add.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
#!/usr/bin/env python2
# -*- coding: utf-8 -*-
import sys
import shutil
import json
from email.parser import HeaderParser
from email.utils import getaddresses
# procmail-py - Email content and spam filtering
# MIT License
# © 2014 Noah K. Tilton <code@tilton.co>
# add.py - programmatically add email addresses to the procmail configuration
# intended to be called from a MUA binding
from config import non_VIPs, VIPs, ADDRESS_FILE, VIPS_FILE
def confirm(msg):
return raw_input("{} [Y/n] ".format(msg)).strip().lower() in ['y', '']
if __name__ == '__main__':
# backup json
[shutil.copyfile(f, f + ".bak") for f in [ADDRESS_FILE, VIPS_FILE] ]
message = HeaderParser().parse(sys.stdin, headersonly=True)
ffroms = message.get_all('from', [])
#ccs = message.get_all('cc', [])
#addrs = getaddresses(ffroms + ccs)
addrs = getaddresses(ffroms)
# reset STDIN handle
sys.stdin = open("/dev/tty")
for name, addr in addrs:
addr = addr.lower()
try:
_ = VIPs[addr]
print ("key exists: {} -> @vip".format( addr ))
continue
except KeyError:
pass
try:
dst = non_VIPs[addr]
print ("key exists: {} -> {}".format( addr, dst ))
continue
except KeyError:
pass
user, domain = addr.split('@')
# prompt for VIP status
fmap = {True: VIPS_FILE, False: ADDRESS_FILE}
vd = {True:'VIP', False:'Non-VIP'}
is_vip = raw_input("{} is VIP [y/N]? ".format(addr)).strip().lower() in ['y']
if is_vip:
folder_name = "@vip"
VIPs.update({addr:folder_name})
if confirm("{} -> {} ({})".format(addr, folder_name, vd[is_vip])):
with open(fmap[is_vip], 'w') as f:
json.dump(VIPs, f)
else:
folder_name = raw_input("{} -> {} [Enter/folder name/X] ".format(addr, user)).strip().lower()
if folder_name == "":
folder_name = user
elif folder_name == "x":
break
if confirm("{} -> {} ({})".format(addr, folder_name, vd[is_vip])):
non_VIPs.update({addr:folder_name})
with open(fmap[is_vip], 'w') as f:
json.dump(non_VIPs, f)