Browse Source

move listing of subscriptions to program

Helmut Pozimski 6 years ago
parent
commit
3d1aed42fc
2 changed files with 26 additions and 14 deletions
  1. 24 0
      lib_stov/program.py
  2. 2 14
      stov

+ 24 - 0
lib_stov/program.py

@@ -98,3 +98,27 @@ def add_subscription(conf, database, channel="", search="", playlist=""):
                                    "database."), video.title)
     LOGGER.info(_("New subscription ") + new_subscription.get_title() +
                 _(" successfully added"))
+
+
+def list_subscriptions(conf, db):
+    """
+    Prints a list of subscriptions from the database.
+
+    :param conf: configuration object
+    :type conf: lib_stov.configuration.Conf
+    :param db: database object
+    :type db: lib_stov.database.Db
+    """
+    subscriptions_list = db.get_subscriptions(conf)
+    sub_state = None
+    if len(subscriptions_list) != 0:
+        LOGGER.info(_("ID Title"))
+        for sub in subscriptions_list:
+            if not sub.disabled:
+                sub_state = _("enabled")
+            elif sub.disabled:
+                sub_state = _("disabled")
+            LOGGER.info(str(sub.get_id()) + " " + sub.get_title() +
+                        " (%s)" % sub_state)
+    else:
+        LOGGER.info(_("No subscriptions added yet, add one!"))

+ 2 - 14
stov

@@ -59,20 +59,8 @@ if ARGUMENTS.add:
     program.add_subscription(CONF, DB, ARGUMENTS.channel,
                              ARGUMENTS.searchparameter,
                              ARGUMENTS.playlist)
-elif ARGUMENTS.list is True:
-    LIST_OF_SUBSCRIPTIONS = DB.get_subscriptions(CONF)
-    SUB_STATE = None
-    if len(LIST_OF_SUBSCRIPTIONS) != 0:
-        LOGGER.info(_("ID Title"))
-        for sub in LIST_OF_SUBSCRIPTIONS:
-            if not sub.disabled:
-                SUB_STATE = _("enabled")
-            elif sub.disabled:
-                SUB_STATE = _("disabled")
-            LOGGER.info(str(sub.get_id()) + " " + sub.get_title() +
-                        " (%s)" % SUB_STATE)
-    else:
-        LOGGER.info(_("No subscriptions added yet, add one!"))
+elif ARGUMENTS.list:
+    program.list_subscriptions(CONF, DB)
 elif ARGUMENTS.deleteid is not None:
     try:
         DELETE_ID = int(ARGUMENTS.deleteid)