list shows in json format

trackma/ui/cli.py

@@ -22,6 +22,7 @@
 import subprocess
 import sys
 import textwrap
+import json
 from operator import itemgetter  # Used for sorting list
 
 from trackma import messenger
@@ -75,6 +76,7 @@ class Trackma_cmd(command.Cmd):
         'mediatype':    (0, 1),
         'info':         1,
         'search':       1,
+        'list_json':    (0, 1),
         'add':          1,
         'del':          1,
         'delete':       1,
@@ -362,6 +364,24 @@ def do_list(self, args):
         # Show the list in memory
         self._make_list(self.sortedlist)
 
+    def do_list_json(self, args):
+        """
+        Print list of shows in json format.
+        Use the command `filter` without arguments to see the available statuses.
+
+        :name list_json
+        :usage list_json <status>
+        """
+        if args:
+            try:
+                self.filter_num = self._guess_status(args[0].lower())
+                self._load_list()
+            except KeyError:
+                print("Invalid filter.")
+                return
+
+        self._make_list_json(self.sortedlist)
+
     def do_info(self, args):
         """
         Gets detailed information about a local show.
@@ -945,6 +965,47 @@ def _make_list(self, showlist):
         print('%d results' % len(showlist))
         print()
 
+    def _make_list_json(self, showlist):
+        """
+        helper function for printing a json formatted show list
+        """
+        altnames = self.engine.altnames()
+
+        episode_str_current = ""
+        episode_str_last = ""
+        # list shows
+        for index, show in showlist:
+            if self.engine.mediainfo['has_progress']:
+                episode_str_current = "{0}".format(show['my_progress'] or '0')
+                episode_str_last = "{0}".format(show['total'] or '?')
+
+            # get title
+            title_str = show['title']
+
+            # ensure score is string; anilist can have string scores such as stars or smiles
+            score = "{0}".format(show['my_score'])
+
+            # json dictionary
+            j = {
+                "title": title_str,
+                "current_episode": episode_str_current,
+                "total_episodes": episode_str_last,
+                "score": score,
+                "airing_status": False,
+                "behind_status": False
+            }
+
+            # Check if show is airing and if user is behind
+            if show['status'] == utils.Status.AIRING:
+                estimate = utils.estimate_aired_episodes(show)
+                if estimate and show['my_progress'] < estimate:
+                    # User is behind the (estimated) aired episode
+                    j["airing_status"] = True;
+                    j["behind_status"] = True;
+                else:
+                    j["airing_status"] = True;
+
+            print(json.dumps(j))
 
 class Trackma_accounts(AccountManager):
     def _get_id(self, index):
@@ -983,12 +1044,12 @@ def select_account(self, bypass):
                     password = getpass.getpass('Enter password (no echo): ')
                 elif selected_api[2] in [utils.Login.OAUTH, utils.Login.OAUTH_PKCE]:
                     username = input('Enter account name: ')
-                    
+
                     auth_url = selected_api[3]
                     if selected_api[2] == utils.Login.OAUTH_PKCE:
                         extra['code_verifier'] = utils.oauth_generate_pkce()
                         auth_url = auth_url % extra['code_verifier']
-                    
+
                     print('OAuth Authentication')
                     print('--------------------')
                     print('This website requires OAuth authentication.')