rain/auto-ttv-grabber
1
0
Fork 0
mirror of https://gitgud.io/Melon__Bread/auto-ttv-grabber.git synced 2024-11-25 00:38:35 -05:00
Code Issues Packages Projects Releases Wiki Activity

Clean up

Browse Source
This commit is contained in:
Rain Clark 2024-04-07 15:47:37 -04:00
parent ed2f65bee5
commit 660b7f1e82
1 changed files with 30 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
main.py
View File

@ -13,34 +13,32 @@ channel_list = []
downloading = {} downloading = {}
# Default Config Settings # 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(): def load_config():
print("Reading config file...") 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"]
print(f"Streamlink location: {streamlink_location}")
if ( if (
config.has_option("settings", "download_location") config.has_option("settings", "download_location")
and not config["settings"]["download_location"].strip() and not config["settings"]["download_location"].strip()
): ):
download_location = config["settings"]["download_location"] download_location = config["settings"]["download_location"]
if ( print(f"Download location: {download_location}")
config.has_option("settings", "skip_ads") if config.has_option("settings", "skip_ads") and not config["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") print("Config file loaded")
else:
print("No config file found using default values!")
def write_log(channel): def write_log(channel):
@ -55,6 +53,7 @@ def download_stream(channel):
"""Downloads a given channel name in its own subprocess""" """Downloads a given channel name in its own subprocess"""
# TODO: Just clean this up at somepoint # TODO: Just clean this up at somepoint
addtional_parms = "" addtional_parms = ""
print(str(skip_ads))
if 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" 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" 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(): def check_system():
"""Makes sure everything is place for the script to run""" """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 # Checks if streamlink is in the systems path
if not shutil.which("streamlink"): if not shutil.which("streamlink"):
sys.exit("ERROR: streamlink is not found in the systems path!") sys.exit("ERROR: streamlink is not found in the systems path!")
@ -141,10 +145,10 @@ def main():
if __name__ == "__main__": if __name__ == "__main__":
""" This is executed when run from the command line """ """This is executed when run from the command line"""
try: try:
load_config()
check_system() check_system()
load_config()
main() main()
finally: finally:
stop_downloads() stop_downloads()