Compare commits

...

6 Commits

Author SHA1 Message Date
91582c67bd Have to use the shell argument for the subprocess 2024-04-06 22:30:51 -04:00
bd874d7245 Hell transfer2 2024-04-06 21:05:40 -04:00
fe61cdf07e Hell transfer 2024-04-06 21:04:41 -04:00
d519900ba8 What is this path issue? 2024-04-05 20:49:48 -04:00
a0ae115748 Fixed f-string 2024-04-05 17:02:58 -04:00
5ee8e47c54 Attempt to switch from dl-stream to streamlink 2024-04-05 16:55:42 -04:00
2 changed files with 23 additions and 11 deletions

View File

@ -1,8 +1,13 @@
# Auto TTV Grabber
<!--toc:start-->
- [Auto TTV Grabber](#auto-ttv-grabber)
- [Getting started](#getting-started)
<!--toc:end-->
## Getting started
This script relies on [dl-stream](https://codeberg.org/bashuser30/dl-stream) being in your systems path.
This script relies on [streamlink](https://streamlink.github.io) being in your systems path.
You can find all of its requirements on the their project page.
This script loops through all of the channels in `channel_list.txt` once per minute checking to seeif the channel is live via a HTTP request.
Once a channel is live a [dl-stream](https://codeberg.org/bashuser30/dl-stream) subprocess spawns in the background downloading the stream to it's default location (`$HOME/Videos/dl-stream/<channel_name>`)
This script loops through all of the channels in `channel_list.txt` once per minute checking to see if the channel is live via a HTTP request.
Once a channel is live a [streamlink](https://streamlink.github.io) subprocess spawns in the background downloading the stream to it's default location (`$HOME/Videos/Stream/<channel_name>`)

23
main.py
View File

@ -7,7 +7,6 @@ import sys
import time
from datetime import datetime
from pathlib import Path
import requests
channel_list = []
@ -16,7 +15,7 @@ downloading = {}
# Default Config Settings
config = configparser.ConfigParser()
streamlink_location = "streamlink"
download_location = str(f"{Path.home()}/Downloads/Streams/")
download_location = f"{Path.home()}/Downloads/Streams"
skip_ads = False
@ -54,8 +53,15 @@ def write_log(channel):
def download_stream(channel):
"""Downloads a given channel name in its own subprocess"""
# TODO: Just clean this up at somepoint
addtional_parms = ""
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"
cmd = f"{streamlink_location} --loglevel none --retry-max 10 {addtional_parms} -o {download_location}/{channel}/{file_name} twitch.tv/{channel} best"
downloading[channel] = subprocess.Popen(
["dl-stream", "-r", channel],
[cmd],
shell=True,
start_new_session=True,
stdout=subprocess.PIPE,
stderr=subprocess.STDOUT,
@ -65,11 +71,11 @@ def download_stream(channel):
def check_system():
"""Makes sure everything is place for the script to run"""
# TODO: Make it fallback to streamlink if dl-stream not present
# Checks if dl-stream is in the systems path
if not shutil.which("dl-stream"):
sys.exit("ERROR: dl-stream is not found in the systems path!")
# Checks if streamlink is in the systems path
if not shutil.which("streamlink"):
sys.exit("ERROR: streamlink is not found in the systems path!")
# TODO: Combine the channel_list into the config
# Checks if the channel_list exists and if not makes one
if not os.path.exists("channel_list.txt"):
print("ERROR:'channel_list.txt' does not exist, creating now!")
@ -110,6 +116,7 @@ def main():
print("\n------------------------------------")
for channel in channel_list:
channel = channel.strip()
# TODO: Have steamlink itself check if the channel is live
contents = requests.get("https://www.twitch.tv/" + channel).content.decode(
"utf-8"
)
@ -118,7 +125,7 @@ def main():
if channel not in downloading:
download_stream(channel)
else:
print("f{channel} is already downloading")
print(f"{channel} is already downloading")
write_log(channel)
else:
print("\033[1m" + channel + "\033[0m is \033[31mnot live\033[0m.")