symuid/symuid/list.py

import argparse
import re
import typing

import symuid


def _walk_track_attrs(path, path_ignore_regex) -> typing.Iterator[dict]:
    for track in symuid.Track.walk(path, path_ignore_regex):
        yield {'path': track.path,
               'comment': track.comment,
               'play_count': track.get_play_count_sum()}

def _init_argparser():
    argparser = argparse.ArgumentParser(description='filter & sort tracks')
    argparser.add_argument('path')
    argparser.add_argument(
        '--path-ignore-regex',
        default=symuid.Track.PATH_DEFAULT_IGNORE_REGEX,
        dest='path_ignore_regex',
        nargs=1,
        type=re.compile,
        metavar='pattern',
        help='(default: %(default)s)',
    )
    argparser.add_argument(
        '--filter',
        metavar='expression',
        dest='filter_expression',
        help='(example: {!r} or {!r})'.format(
            "play_count > 16 and path.endswith('.mp3')",
            "comment is None or len(comment) < 16",
        ),
    )
    argparser.add_argument(
        '--sort',
        metavar='expression',
        dest='sort_expression',
        help='(example: {!r} or {!r})'.format(
            "play_count * -1",
            "(play_count, len(path))",
        ),
    )
    argparser.add_argument(
        '--limit',
        type=int,
        help='(default: none)',
    )
    argparser.add_argument(
        '--prefix',
        type=str,
        default='',
        help='add prefix to each resulting path (default: %(default)r)',
    )
    return argparser


def _main():
    args = _init_argparser().parse_args()
    # use generators until sort is required
    track_attrs = _walk_track_attrs(path=args.path, path_ignore_regex=args.path_ignore_regex)
    # pylint: disable=eval-used
    if args.filter_expression:
        track_attrs = filter(lambda a: eval(args.filter_expression, a), track_attrs)
    if args.sort_expression:
        track_attrs = sorted(track_attrs, key=lambda a: eval(args.sort_expression, a))
    for track_index, attr in enumerate(track_attrs):
        if args.limit and track_index == args.limit:
            break
        print(args.prefix + attr['path'])