test.cc: new C++ test

[home.git] / .config / mutt / list-mailboxes.py

#!/usr/bin/python

"""Produce a "sorted" list of maildir paths based on subdirs."""

from __future__ import print_function

import os
import sys


IGNORED_DIRS = set((
    'drafts',
    'outbox',
    'templates',
    '.inbox.directory',
))


def find_dirs(topdir, ignored):
    """Walk the subdirs following the KDE layout."""
    ret = []
    for d in sorted(os.listdir(topdir)):
        if d in ('cur', 'new', 'tmp'):
            continue

        fulld = os.path.join(topdir, d)
        if d.startswith('.') or fulld in ignored:
            continue
        if not os.path.islink(fulld):
            if os.path.isdir(os.path.join(fulld, 'cur')):
                ret.append(fulld)
                subdir = os.path.join(topdir, '.%s.directory' % d)
                if subdir not in ignored and os.path.isdir(subdir):
                    ret.extend(find_dirs(subdir, ignored))

    return ret


def main(argv):
    maildir = os.path.expanduser('~/.mail/')
    results = []
    ignored = set(os.path.join(maildir, x) for x in IGNORED_DIRS)

    results = find_dirs(maildir, ignored)
    print(' '.join('"%s"' % x for x in results))
    return 0


# Old logic that doesn't sort right.
    for root, dirs, _ in os.walk(maildir, followlinks=False):
        for d in ('cur', 'new', 'tmp'):
            try:
                dirs.remove(d)
            except ValueError:
                pass

        dirs.sort()
        for d in dirs[:]:
            fulld = os.path.join(root, d)
            if fulld in ignored:
                dirs.remove(d)
            elif not os.path.islink(fulld):
                if os.path.isdir(os.path.join(fulld, 'cur')):
                    results.append(fulld)

    print('\n'.join('"%s"' % x for x in results))


if __name__ == '__main__':
    sys.exit(main(sys.argv[1:]))