add option to manually set theme

automathemely/autoth_tools/argmanager.py

@@ -19,6 +19,10 @@
 options.add_argument('-r', '--restart',
                      help='(re)start the scheduler script if it were to not start or stop unexpectedly',
                      action='store_true', default=False)
+options.add_argument('--light',
+                     help='manually set current theme to Light', action='store_true', default=False)
+options.add_argument('--dark',
+                     help='manually set current theme to Dark', action='store_true', default=False)
 
 
 #   For --list arg
@@ -119,3 +123,10 @@ def main(us_se):
             Popen(['pkill', '-f', 'autothscheduler.py']).wait()
         Popen(['python3', get_bin('autothscheduler.py')], start_new_session=True, stdout=DEVNULL, stderr=DEVNULL)
         logger.info('Restarted the scheduler')
+
+    #   MANUALLY SET THEME
+    elif args.light:
+        return "light"
+    elif args.dark:
+        return "dark"
+

automathemely/bin/run.py

@@ -82,9 +82,11 @@ def main():
         logging.getLogger().addHandler(automathemely.notifier_handler)
 
     #   If any argument is given, pass it/them to the arg manager module
+    manual_theme_set = None
     if len(sys.argv) > 1:
-        automathemely.autoth_tools.argmanager.main(user_settings)
-        return
+        manual_theme_set = automathemely.autoth_tools.argmanager.main(user_settings)
+        if manual_theme_set is None:
+            return
 
     # We don't want to proceed until we have given the user the chance to review its settings
     if first_time_run:
@@ -111,6 +113,9 @@ def main():
     else:
         t_color = 'dark'
 
+    if manual_theme_set is not None:
+        t_color = manual_theme_set
+
     logger.info('Switching to {} themes...'.format(t_color))
 
     #   Change desktop environment theme