Clean up

main.py

@@ -1,46 +1,44 @@
 #!/usr/bin/env python3
 import configparser
 import os
-import requests
 import shutil
 import subprocess
 import sys
 import time
 from datetime import datetime
 from pathlib import Path
+import requests
 
 channel_list = []
 downloading = {}
 
 # Default Config Settings
-config = configparser.ConfigParser()
+streamlink_location: str = "streamlink"
-streamlink_location = "streamlink"
+download_location: str = f"{Path.home()}/Downloads/Streams"
-download_location = f"{Path.home()}/Downloads/Streams"
+skip_ads: bool = False
-skip_ads = False
 
 
 def load_config():
     print("Reading config file...")
-    if os.path.exists("config.ini"):
+    config = configparser.ConfigParser()
-        config.read("config.ini")
+    config.read("config.ini")
-        if (
+    if (
-            config.has_option("settings", "streamlink_location")
+        config.has_option("settings", "streamlink_location")
-            and not config["settings"]["streamlink_location"].strip()
+        and not config["settings"]["streamlink_location"].strip()
-        ):
+    ):
-            streamlink_location = config["settings"]["streamlink_location"]
+        streamlink_location = config["settings"]["streamlink_location"]
-        if (
+        print(f"Streamlink location: {streamlink_location}")
-            config.has_option("settings", "download_location")
+    if (
-            and not config["settings"]["download_location"].strip()
+        config.has_option("settings", "download_location")
-        ):
+        and not config["settings"]["download_location"].strip()
-            download_location = config["settings"]["download_location"]
+    ):
-        if (
+        download_location = config["settings"]["download_location"]
-            config.has_option("settings", "skip_ads")
+        print(f"Download location: {download_location}")
-            and not config["settings"]["skip_ads"]
+    if config.has_option("settings", "skip_ads") and not config["settings"]["skip_ads"]:
-        ):
+        skip_ads = bool(config["settings"]["skip_ads"])
-            skip_ads = bool(config["settings"]["skip_ads"])
+        print(f"Skip ads: {skip_ads}")
-        print("Config file loaded")
+
-    else:
+    print("Config file loaded")
-        print("No config file found using default values!")
 
 
 def write_log(channel):
@@ -55,6 +53,7 @@ def download_stream(channel):
     """Downloads a given channel name in its own subprocess"""
     # TODO: Just clean this up at somepoint
     addtional_parms = ""
+    print(str(skip_ads))
     if skip_ads:
         addtional_parms = "--twitch-proxy-playlist=https://lb-eu.cdn-perfprod.com,https://lb-eu2.cdn-perfprod.com,https://lb-na.cdn-perfprod.com,https://lb-as.cdn-perfprod.com,https://as.luminous.dev --twitch-disable-ads"
     file_name = f"{channel}_{datetime.now().strftime('%Y-%m-%d_%H-%M-%S')}.ts"
@@ -71,6 +70,11 @@ def download_stream(channel):
 
 def check_system():
     """Makes sure everything is place for the script to run"""
+
+    # Checks for config file
+    if not os.path.exists("config.ini"):
+        sys.exit("ERROR: config.ini is not found! See README.md for more info.")
+
     # Checks if streamlink is in the systems path
     if not shutil.which("streamlink"):
         sys.exit("ERROR: streamlink is not found in the systems path!")
@@ -141,10 +145,10 @@ def main():
 
 
 if __name__ == "__main__":
-    """ This is executed when run from the command line """
+    """This is executed when run from the command line"""
     try:
-        load_config()
         check_system()
+        load_config()
         main()
     finally:
         stop_downloads()