symuid/symuid/import_library/itunes_xml.py

import argparse
import os
import re
import sys

import symuid
import symuid.library.itunes

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


def import_itunes_xml_library(xml_library_path, path_regex_sub):
    lib = symuid.library.itunes.XmlLibrary(xml_library_path)
    for itunes_track in lib.tracks:
        # TODO create tag if last_play_dt is None
        if itunes_track.last_play_dt and itunes_track.local:
            track_path = itunes_track.local_path
            for pattern, repl in path_regex_sub:
                track_path = re.sub(pattern, repl, track_path)
            if not os.path.exists(track_path):
                sys.stderr.write("{!r}: not found\n".format(track_path))
            else:
                symuid_track = symuid.Track(path=track_path)
                # TODO dt=dt.datetime.now()
                symuid_track.register_play_count(
                    symuid.PlayCount(
                        player="itunes",
                        library_id=lib.library_id,
                        register_dt=itunes_track.last_play_dt,
                        count=itunes_track.play_count,
                    ),
                    tag_set_cb=lambda tr, tag: print(
                        "{!r}: set tag {!r}".format(tr.path, tag)
                    ),
                )


def _main():
    argparser = argparse.ArgumentParser(
        description="Import play counts from iTunes XML Library"
    )
    argparser.add_argument("xml_library_path")
    argparser.add_argument(
        "--path-regex-sub",
        nargs=2,
        action="append",
        metavar=("regex", "replacement"),
        default=[],
        help="(default: no substitution)",
    )
    args = argparser.parse_args()
    import_itunes_xml_library(**vars(args))