symuid/symuid/import_library/cmus_cache.py

import argparse
import os
import sys

import symuid
import symuid.library.cmus
from symuid._datetime import datetime_utc_now

# TODO rename module to cmus.py
# after ImportMismatchError on `pytest --doctest-modules` was fixed

_PLAYER_NAME = "cmus"
_LIBRARY_ID_LENGTH_MIN = 8


def symuid_import_cmus(library_id, cache_path):
    assert len(library_id) >= _LIBRARY_ID_LENGTH_MIN, library_id
    lib = symuid.library.cmus.Cache(cache_path)
    for cmus_track in lib.get_tracks():
        if not os.path.exists(cmus_track.path):
            sys.stderr.write("{!r}: not found\n".format(cmus_track.path))
        elif (
            cmus_track.play_count > 0
        ):  # check before symuid.Track to improve performance
            symuid_track = symuid.Track(path=cmus_track.path.decode())
            last_count = symuid_track.get_latest_play_count(
                player=_PLAYER_NAME, library_id=library_id,
            )
            assert last_count is None or last_count.count <= cmus_track.play_count, (
                symuid_track.path,
                last_count.count,
                cmus_track.play_count,
            )
            if last_count is None or last_count.count != cmus_track.play_count:
                symuid_track.register_play_count(
                    symuid.PlayCount(
                        player=_PLAYER_NAME,
                        library_id=library_id,
                        register_dt=datetime_utc_now(),
                        count=cmus_track.play_count,
                    ),
                    tag_set_cb=lambda track, tag: print(
                        "{!r}: set tag {!r}".format(track.path, tag)
                    ),
                )


def _main():
    argparser = argparse.ArgumentParser(
        description="Import play counts from cmus' cache"
    )
    argparser.add_argument("library_id")
    argparser.add_argument(
        "cache_path",
        nargs="?",
        default=os.path.expanduser("~/.config/cmus/cache"),
        help="(default: %(default)r)",
    )
    args = argparser.parse_args()
    symuid_import_cmus(**vars(args))