Browse Source

setup `symuid-import-itunes` cli entry point via setuptools.setup()

Fabian Peter Hammerle 4 years ago
parent
commit
c4c910af50
2 changed files with 9 additions and 15 deletions
  1. 1 0
      setup.py
  2. 8 15
      symuid/import_library/itunes_xml.py

+ 1 - 0
setup.py

@@ -6,6 +6,7 @@ setuptools.setup(
     packages=setuptools.find_packages(),
     entry_points={
         'console_scripts': [
+            'symuid-import-itunes = symuid.import_library.itunes_xml:_main',
             'symuid-sync = symuid.sync:_main',
         ],
     },

+ 8 - 15
symuid-import-itunes → symuid/import_library/itunes_xml.py

@@ -1,15 +1,13 @@
-#!/usr/bin/env python3
-# -*- coding: utf-8 -*-
-
-import datetime as dt
+import argparse
 import os
 import re
+import sys
+
 import symuid
 import symuid.library.itunes
-import sys
 
 
-def symuid_import_itunes(xml_library_path, path_regex_sub):
+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
@@ -30,12 +28,11 @@ def symuid_import_itunes(xml_library_path, path_regex_sub):
                         count=itunes_track.play_count,
                     ),
                     tag_set_cb=lambda tr, tag:
-                        print('{!r}: set tag {!r}'.format(tr.path, tag)),
+                    print('{!r}: set tag {!r}'.format(tr.path, tag)),
                 )
 
 
 def _init_argparser():
-    import argparse
     argparser = argparse.ArgumentParser(description=None)
     argparser.add_argument('xml_library_path')
     argparser.add_argument(
@@ -49,11 +46,7 @@ def _init_argparser():
     return argparser
 
 
-def main(argv):
+def _main():
     argparser = _init_argparser()
-    args = argparser.parse_args(argv[1:])
-    symuid_import_itunes(**vars(args))
-    return 0
-
-if __name__ == "__main__":
-    sys.exit(main(sys.argv))
+    args = argparser.parse_args()
+    import_itunes_xml_library(**vars(args))