play: allow selecting a specific season of media

Author: TAAPArthur

amt/args.py

@@ -100,6 +100,7 @@ def setup_subparsers(state, sub_parsers):
     sub_consume_parsers.add_argument("--shuffle", "-s", default=False, action="store_const", const=True)
     sub_consume_parsers.add_argument("--stream-index", "-q", default=0, type=int)
     sub_consume_parsers.add_argument("--volume", default=False, action="store_const", const=True)
+    sub_consume_parsers.add_argument("--season", dest="season_number", type=int)
     sub_consume_parsers.add_argument("--raw", default=False, action="store_const", const=True)
     sub_consume_parsers.add_argument("--special", default=None, action="store_const", const=True)
     sub_consume_parsers.add_argument("--nospecial", default=None, dest="special", action="store_const", const=False)

amt/media_reader.py

@@ -471,13 +471,15 @@ def maybe_resolve_media_type(self, media_data, media_type_filter=None):
                 media_data["media_type"] = media_type.value
                 media_data["media_type_name"] = media_type.name
 
-    def play(self, name=None, media_type=None, shuffle=False, limit=None, num_list=None, stream_index=0, raw=False, special=None, any_unread=False, force_abs=False, force_stream=False, volume=False, batch_size=1, **download_chapter_args):
+    def play(self, name=None, media_type=None, shuffle=False, limit=None, num_list=None, season_number=None, stream_index=0, raw=False, special=None, any_unread=False, force_abs=False, force_stream=False, volume=False, batch_size=1, **download_chapter_args):
         num = 0
         batch = []
         server_media_chapters = []
         for info in self.get_chapters(name, media_type=media_type, num_list=num_list, limit=limit, shuffle=shuffle, any_unread=any_unread, force_abs=force_abs, volume=volume, null_terminate=True):
             if info:
                 server, media_data, chapter = info
+                if season_number is not None and chapter["volume_number"] != season_number:
+                    continue
                 if special is not None and chapter["special"] != special:
                     continue
                 self.maybe_resolve_media_type(media_data, media_type)