Added Environment Variables #107

plex_meta_manager.py

@@ -15,23 +15,33 @@ parser.add_argument("-c", "--config", dest="config", help="Run with desired *.ym
 parser.add_argument("-t", "--time", dest="time", help="Time to update each day use format HH:MM (Default: 03:00)", default="03:00", type=str)
 parser.add_argument("-r", "--run", dest="run", help="Run without the scheduler", action="store_true", default=False)
 parser.add_argument("-rt", "--test", "--tests", "--run-test", "--run-tests", dest="test", help="Run in debug mode with only collections that have test: true", action="store_true", default=False)
-parser.add_argument("-cl", "--collection", "--collections", dest="collections", help="Process only specified collections (comma-separated list)", type=str, default="")
+parser.add_argument("-cl", "--collection", "--collections", dest="collections", help="Process only specified collections (comma-separated list)", type=str)
 parser.add_argument("-d", "--divider", dest="divider", help="Character that divides the sections (Default: '=')", default="=", type=str)
 parser.add_argument("-w", "--width", dest="width", help="Screen Width (Default: 100)", default=100, type=int)
 args = parser.parse_args()
 
-if not re.match("^([0-1]?[0-9]|2[0-3]):[0-5][0-9]$", args.time):
+my_tests = os.environ.get("PMM_TESTS") if os.environ.get("PMM_TESTS") else args.tests
-    raise util.Failed(f"Argument Error: time argument invalid: {args.time} must be in the HH:MM format")
+test = os.environ.get("PMM_TEST") if os.environ.get("PMM_TEST") else args.test
+debug = os.environ.get("PMM_DEBUG") if os.environ.get("PMM_DEBUG") else args.debug
+run = os.environ.get("PMM_RUN") if os.environ.get("PMM_RUN") else args.run
+collections = os.environ.get("PMM_COLLECTIONS") if os.environ.get("PMM_COLLECTIONS") else args.collections
 
-util.separating_character = args.divider[0]
+time_to_run = os.environ.get("PMM_TIME") if os.environ.get("PMM_TIME") else args.time
-if 90 <= args.width <= 300:
+if not re.match("^([0-1]?[0-9]|2[0-3]):[0-5][0-9]$", time_to_run):
-    util.screen_width = args.width
+    raise util.Failed(f"Argument Error: time argument invalid: {time_to_run} must be in the HH:MM format")
+
+util.separating_character = os.environ.get("PMM_DIVIDER")[0] if os.environ.get("PMM_DIVIDER") else args.divider[0]
+
+screen_width = os.environ.get("PMM_WIDTH") if os.environ.get("PMM_WIDTH") else args.width
+if 90 <= screen_width <= 300:
+    util.screen_width = screen_width
 else:
-    raise util.Failed(f"Argument Error: width argument invalid: {args.width} must be an integer between 90 and 300")
+    raise util.Failed(f"Argument Error: width argument invalid: {screen_width} must be an integer between 90 and 300")
 
+config_file = os.environ.get("PMM_CONFIG") if os.environ.get("PMM_CONFIG") else args.config
 default_dir = os.path.join(os.path.dirname(os.path.abspath(__file__)), "config")
-if args.config and os.path.exists(args.config):                     default_dir = os.path.join(os.path.dirname(os.path.abspath(args.config)))
+if config_file and os.path.exists(config_file):                     default_dir = os.path.join(os.path.dirname(os.path.abspath(config_file)))
-elif args.config and not os.path.exists(args.config):               raise util.Failed(f"Config Error: config not found at {os.path.abspath(args.config)}")
+elif config_file and not os.path.exists(config_file):               raise util.Failed(f"Config Error: config not found at {os.path.abspath(config_file)}")
 elif not os.path.exists(os.path.join(default_dir, "config.yml")):   raise util.Failed(f"Config Error: config not found at {os.path.abspath(default_dir)}")
 
 os.makedirs(os.path.join(default_dir, "logs"), exist_ok=True)
@@ -50,7 +60,7 @@ file_handler.setFormatter(logging.Formatter("[%(asctime)s] %(filename)-27s %(lev
 
 cmd_handler = logging.StreamHandler()
 cmd_handler.setFormatter(logging.Formatter("| %(message)-100s |"))
-cmd_handler.setLevel(logging.DEBUG if args.tests or args.test or args.debug else logging.INFO)
+cmd_handler.setLevel(logging.DEBUG if tests or test or debug else logging.INFO)
 
 logger.addHandler(cmd_handler)
 logger.addHandler(file_handler)
@@ -68,20 +78,20 @@ logger.info(util.get_centered_text("
 logger.info(util.get_centered_text("    Version: 1.5.0                                                                   "))
 util.separator()
 
-if args.tests:
+if my_tests:
     tests.run_tests(default_dir)
     sys.exit(0)
 
-def start(config_path, test, daily, collections):
+def start(config_path, is_test, daily, collections):
     if daily:               start_type = "Daily "
-    elif test:              start_type = "Test "
+    elif is_test:           start_type = "Test "
     elif collections:       start_type = "Collections "
     else:                   start_type = ""
     start_time = datetime.now()
     util.separator(f"Starting {start_type}Run")
     try:
         config = Config(default_dir, config_path)
-        config.update_libraries(test, collections)
+        config.update_libraries(is_test, collections)
     except Exception as e:
         util.print_stacktrace()
         logger.critical(e)
@@ -89,15 +99,15 @@ def start(config_path, test, daily, collections):
     util.separator(f"Finished {start_type}Run\nRun Time: {str(datetime.now() - start_time).split('.')[0]}")
 
 try:
-    if args.run or args.test or args.collections:
+    if run or test or collections:
-        start(args.config, args.test, False, args.collections)
+        start(config_file, test, False, collections)
     else:
         length = 0
-        schedule.every().day.at(args.time).do(start, args.config, False, True, None)
+        schedule.every().day.at(time_to_run).do(start, config_file, False, True, None)
         while True:
             schedule.run_pending()
             current = datetime.now().strftime("%H:%M")
-            seconds = (datetime.strptime(args.time, "%H:%M") - datetime.strptime(current, "%H:%M")).total_seconds()
+            seconds = (datetime.strptime(time_to_run, "%H:%M") - datetime.strptime(current, "%H:%M")).total_seconds()
             hours = int(seconds // 3600)
             if hours < 0:
                 hours += 24
@@ -105,7 +115,7 @@ try:
             time_str = f"{hours} Hour{'s' if hours > 1 else ''} and " if hours > 0 else ""
             time_str += f"{minutes} Minute{'s' if minutes > 1 else ''}"
 
-            length = util.print_return(length, f"Current Time: {current} | {time_str} until the daily run at {args.time}")
+            length = util.print_return(length, f"Current Time: {current} | {time_str} until the daily run at {time_to_run}")
             time.sleep(1)
 except KeyboardInterrupt:
     util.separator("Exiting Plex Meta Manager")