added specific playlist refreshing

generateplaylists.py

@@ -10,44 +10,35 @@ from spotframework.engine.filter.sortreversereleasedate import SortReverseReleas
 from spotframework.engine.filter.deduplicatebyid import DeduplicateByID
 from spotframework.engine.filter.deduplicatebyname import DeduplicateByName
 
+import os
 import datetime
+import sys
 
 import requests
 
-import os
 
-if __name__ == '__main__':
+def update_super_playlist(engine, data_dict):
 
-    try:
+    log.log("makePlaylist", data_dict['name'])
-        if os.path.exists(os.path.join(const.config_path, 'playlists.json')):
-
-            data = json.load_json(os.path.join(const.config_path, 'playlists.json'))
-
-            net = Network(User())
-
-            engine = PlaylistEngine(net)
-            engine.load_user_playlists()
-
-            for tomake in data['playlists']:
-
-                log.log("makePlaylist", tomake['name'])
 
     processors = [DeduplicateByID()]
 
-                if 'shuffle' in tomake:
+    if 'shuffle' in data_dict:
-                    if tomake['shuffle'] is True:
+        if data_dict['shuffle'] is True:
             processors.append(Shuffle())
         else:
             processors.append(SortReverseReleaseDate())
     else:
         processors.append(SortReverseReleaseDate())
 
-                tracks = engine.make_playlist(tomake['playlists'], processors)
+    tracks = engine.make_playlist(data_dict['playlists'], processors)
 
-                engine.execute_playlist(tracks, tomake['id'])
+    engine.execute_playlist(tracks, data_dict['id'])
-                engine.change_description(tomake['playlists'], tomake['id'])
+    engine.change_description(data_dict['playlists'], data_dict['id'])
+
+
+def update_recents_playlist(engine, data):
 
-            if 'recents' in data:
     recents_id = data['recents']['id']
     boundary_date = datetime.datetime.now() - datetime.timedelta(days=data['recents']['boundary'])
 
@@ -70,10 +61,55 @@ if __name__ == '__main__':
     processors = [DeduplicateByName(), SortReverseReleaseDate()]
 
     recent_tracks = engine.get_recent_playlist(boundary_date, recent_parts, processors)
-                engine.execute_playlist(recent_tracks, data['recents']['id'])
+    engine.execute_playlist(recent_tracks, recents_id)
-                engine.change_description([monthstrings.get_this_month(),
+    engine.change_description([monthstrings.get_this_month(), monthstrings.get_last_month()], data['recents']['id'])
-                                           monthstrings.get_last_month()]
+
-                                          , data['recents']['id'])
+
+def go():
+
+    try:
+        if os.path.exists(os.path.join(const.config_path, 'playlists.json')):
+
+            data = json.load_json(os.path.join(const.config_path, 'playlists.json'))
+
+            to_execute = []
+            not_found = []
+
+            available_specials = ['recents']
+            specials_to_execute = []
+
+            if len(sys.argv) > 1:
+                for arg in sys.argv[1:]:
+                    for playlist in data['playlists']:
+                        if arg.lower() == playlist['name'].lower():
+                            to_execute.append(playlist)
+                            break
+                    else:
+                        if arg in available_specials:
+                            specials_to_execute.append(arg)
+                        else:
+                            not_found.append(arg)
+            else:
+                to_execute = data['playlists']
+                specials_to_execute = ['recents']
+
+            if len(not_found) > 0:
+                log.log('arg not found', not_found)
+
+            if len(to_execute) <= 0 and len(specials_to_execute) <= 0:
+                log.log('none to execute, terminating')
+                return
+
+            net = Network(User())
+
+            engine = PlaylistEngine(net)
+            engine.load_user_playlists()
+
+            for super_playlist in to_execute:
+                update_super_playlist(engine, super_playlist)
+
+            if 'recents' in data and 'recents' in specials_to_execute:
+                update_recents_playlist(engine, data)
 
         else:
             log.log("config json not found")
@@ -86,3 +122,7 @@ if __name__ == '__main__':
         if 'SLACKHOOK' in os.environ:
             requests.post(os.environ['SLACKHOOK'], json={"text": "spot playlists: exception occured"})
         log.dump_log()
+
+
+if __name__ == '__main__':
+    go()