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
|
#!/usr/bin/env python3
import imaplib
import sys
def get_boxes(list):
"""
Retrieve and decode all mailboxes.
"""
return [box.decode('utf-8').rsplit(' ')[-1] for box in list]
def flatten_output(list):
"""
Print all boxes with a flattened output,
primarily for neomutt usage.
"""
print(''.join([f"+'{box}' " for box in list]), end=' ')
def sort_boxes(bxs):
"""
Sort boxes list,
according to a predefined order
"""
order = [
"INBOX",
"Unread",
"Drafts",
"Sent",
"Spam",
"Trash",
"Junk",
"Archive"
]
bxs_orig = sorted(bxs)
# sort based on predefined order
bxs = []
for exp in order:
matching = [s for s in bxs_orig if exp in s]
bxs.extend(matching)
# ensure all retrieved boxes are present
for bx in bxs_orig:
if bx not in bxs:
bxs.append(bx)
return bxs
def main():
"""
Retrieve, sort, and pretty print for neomutt
"""
# user information
remote = sys.argv[1]
username = sys.argv[2]
password = sys.argv[3]
# connection
mail = imaplib.IMAP4_SSL(remote)
mail.login(username, password)
# parse folders output
bxs = sort_boxes(get_boxes(mail.list()[1]))
# oneline pretty-print for neomutt
flatten_output(bxs)
if __name__ == "__main__":
main()
|
