You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
84 lines
1.9 KiB
Python
84 lines
1.9 KiB
Python
#!/usr/bin/env python
|
|
"""
|
|
List maildirs in your Maildir++ with message counts.
|
|
"""
|
|
from __future__ import division, print_function
|
|
|
|
import os
|
|
import os.path
|
|
import re
|
|
import yaml
|
|
|
|
def maildirs(startdir):
|
|
for dirpath, dirnames, _ in os.walk(startdir):
|
|
if set(['cur', 'new', 'tmp']).issubset(set(dirnames)):
|
|
yield dirpath
|
|
|
|
|
|
def maildir_count(maildir):
|
|
|
|
def dir_count(dir_):
|
|
return len(os.listdir(dir_))
|
|
|
|
cur = dir_count(os.path.join(maildir, 'cur'))
|
|
new = dir_count(os.path.join(maildir, 'new'))
|
|
count = cur + new
|
|
|
|
return (new, count)
|
|
|
|
|
|
def mailbox_name(maildir, root):
|
|
name = re.sub(root + r'/?\.?', '', maildir)
|
|
if name == '':
|
|
return 'INBOX'
|
|
else:
|
|
return name
|
|
|
|
|
|
config_filename = '~/.config/maildir-zero.yml'
|
|
config = {}
|
|
try:
|
|
config = yaml.load(open(os.path.expanduser(config_filename)))
|
|
except:
|
|
pass
|
|
ignore_zero = config.get('ignore_zero', True)
|
|
ignore = config.get('ignore', [r'spam'])
|
|
sort_by_count = config.get('sort_by_count', True)
|
|
|
|
|
|
root = os.path.expanduser('~/Maildir')
|
|
counts = {}
|
|
for maildir in maildirs(root):
|
|
name = mailbox_name(maildir, root)
|
|
_, count = maildir_count(maildir)
|
|
|
|
if count == 0 and ignore_zero:
|
|
continue
|
|
|
|
if any(re.match(i, name) for i in ignore):
|
|
continue
|
|
|
|
counts[name] = count
|
|
|
|
if sort_by_count:
|
|
key = lambda i: i[1]
|
|
reverse = True
|
|
else:
|
|
key = lambda i: i[0]
|
|
reverse = False
|
|
|
|
try:
|
|
length_name = max(len(name) for name in counts.keys())
|
|
except:
|
|
length_name = len('Total')
|
|
try:
|
|
length_count = max(len(str(count)) for count in counts.values())
|
|
except:
|
|
length_count = 3
|
|
|
|
for name, count in sorted(counts.items(), key=key, reverse=reverse):
|
|
print('{0:{1}}\t{2:{3}d}'.format(name, length_name, count, length_count))
|
|
|
|
total = sum(counts.values())
|
|
print('\n{0:{1}}\t{2:{3}d}'.format('Total', length_name, total, length_count))
|