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.

77 lines
1.8 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'))
all_ = cur + new
return (new, all_)
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)
_, all_ = maildir_count(maildir)
if all_ == 0 and ignore_zero:
continue
if any(re.match(i, name) for i in ignore):
continue
counts[name] = all_
if sort_by_count:
key = lambda i: i[1]
reverse = True
else:
key = lambda i: i[0]
reverse = False
length_name = max(len(name) for name in counts.keys())
length_count = max(len(str(count)) for count in counts.values())
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))