autogenerate category map

Author: mueslo

pythonbits/__main__.py

@@ -5,7 +5,7 @@
 from . import __version__ as version
 from . import bb
 from . import logging
-from .submission import SubmissionAttributeError
+from .submission import SubmissionAttributeError, cat_map
 
 
 def parse_args():
@@ -22,9 +22,6 @@ def parse_args():
                               "(e.g. \"Lawrence of Arabia\" or \"The Walking "
                               "Dead S01\") (optional)"))
 
-    cat_map = {'movie': bb.MovieSubmission,
-               'tv': bb.TvSubmission,
-               'music': bb.MusicSubmission}
     parser.add_argument("-c", "--category", choices=list(cat_map.keys()))
     parser.add_argument("-u", "--set-field", nargs=2, action='append',
                         metavar=('FIELD', 'VALUE'), default=[],

pythonbits/bb.py

@@ -26,7 +26,7 @@
 from . import imagehosting
 from .ffmpeg import FFMpeg
 from . import templating as bb
-from .submission import (Submission, form_field, finalize,
+from .submission import (Submission, form_field, finalize, cat_map,
                          SubmissionAttributeError, rlinput)
 from .tracker import Tracker
 from .scene import is_scene_crc, query_scene_fname
@@ -81,9 +81,15 @@ def subcategory(self):
                 elif mime_guess[0] == 'audio':
                     return AudioSubmission
 
-        raise Exception("Unable to guess category using known mimetypes. "
-                        "Consider explicitly specifying the submission "
-                        "category.")
+        log.info("Unable to guess submission category using known mimetypes")
+        while True:
+            cat = input("Please manually specify category. "
+                        "\nOptions: {}\nCategory: ".format(
+                ", ".join(cat_map.keys())))
+            try:
+                return cat_map[cat]
+            except KeyError:
+                print('Invalid category.')
 
     def subcategorise(self):
         log.debug('Attempting to narrow category')
@@ -571,6 +577,7 @@ def _finalize_cover(self):
 
 class TvSubmission(VideoSubmission):
     default_fields = VideoSubmission.default_fields + ('form_description',)
+    _cat_id = 'tv'
     _form_type = 'TV'
     __form_fields__ = {
         'form_title': ('title', 'text'),
@@ -781,6 +788,7 @@ def episode_fmt(e):
 class MovieSubmission(VideoSubmission):
     default_fields = (VideoSubmission.default_fields +
                       ("description", "mediainfo", "screenshots"))
+    _cat_id = 'movie'
     _form_type = 'Movies'
     __form_fields__ = {
         # field -> form field, type
@@ -1173,6 +1181,7 @@ def _get_tags(self, required_tags):
 
 
 class AudiobookSubmission(AudioSubmission):
+    _cat_id = 'audiobook'
     _form_type = 'Audiobooks'
 
     @form_field('tags')
@@ -1189,6 +1198,7 @@ def _render_title(self):
 class MusicSubmission(AudioSubmission):
     default_fields = (AudioSubmission.default_fields + (
          'artist', 'remaster', 'remaster_year', 'remaster_title', 'media',))
+    _cat_id = 'music'
     _form_type = 'Music'
 
     @form_field('remaster_true', 'checkbox')

pythonbits/submission.py

@@ -25,13 +25,17 @@ class SubmissionAttributeError(Exception):
 
 
 re_frender = re.compile("^_render_(?=[a-z_]*$)")
+cat_map = {}
 
 
 class RegisteringType(type):
     def __init__(cls, name, bases, attrs):
         cls.registry = copy.deepcopy(getattr(cls, 'registry',
                                              {'mappers': {}, 'types': {}}))
 
+        if hasattr(cls, '_cat_id'):
+            cat_map[cls._cat_id] = cls
+
         def add_mapper(f, ff, fft):
             log.debug("{} adding mapper {} for {} ({})",
                       cls.__name__, f, ff, fft)