diff options
| -rw-r--r-- | stashley/stashley.py | 97 |
1 files changed, 64 insertions, 33 deletions
diff --git a/stashley/stashley.py b/stashley/stashley.py index 16af706..94538ef 100644 --- a/stashley/stashley.py +++ b/stashley/stashley.py @@ -32,51 +32,80 @@ def legalfy(url: str) -> str: return ''.join(letter for letter in url if letter.isalnum()) -def figureUse(lore: dict, keyname: str) -> str: +def figureUse(lore: dict, keyname: str) -> dict: + + normopts = {'allsubtitles': True, + 'extract_flat': 'discard_in_playlist', + 'format': 'bestvideo[height>=720][fps>=60]+bestaudio/bestvideo+bestaudio/best', + 'fragment_retries': 10, + 'ignoreerrors': True, + 'outtmpl': {'default': keyname + '/%(uploader|author)s/%(title)s.%(ext)s'}, + 'postprocessors': [ + {'key': 'FFmpegConcat', + 'only_multi_video': True, + 'when': 'playlist'}], + 'retries': 10, + 'writeannotations': True, + 'writedescription': True, + 'writeinfojson': True, + 'writesubtitles': True, + 'writethumbnail': True} + + tiktokopts = {'allsubtitles': True, + 'extract_flat': 'discard_in_playlist', + 'format': 'bestvideo[height>=720][fps>=60]+bestaudio/bestvideo+bestaudio/best', + 'fragment_retries': 10, + 'ignoreerrors': True, + 'outtmpl': {'default': keyname + '/%(uploader|author)s/%(title)s.%(ext)s'}, + 'postprocessors': [ + {'key': 'FFmpegConcat', + 'only_multi_video': True, + 'when': 'playlist'}, + {'key': 'FFmpegVideoConvertor', + 'preferedformat': 'webm'} + ], + 'retries': 10, + 'writeannotations': True, + 'writedescription': True, + 'writeinfojson': True, + 'writesubtitles': True, + 'writethumbnail': True} + + + + if "use" in lore: match lore["use"]: case "yt-dlp": - return "ytdl" + return {"tool": "ytdl", "opts": normopts} case "ytdlp": - return "ytdl" + return {"tool": "ytdl", "opts": normopts} case "youtube-dl": - return "ytdl" + return {"tool": "ytdl", "opts": normopts} case "yt-dl": - return "ytdl" + return {"tool": "ytdl", "opts": normopts} case "website": - return "website" + return {"tool": "website"} + case "websites": + return {"tool": "website"} case "site": - return "website" + return {"tool": "website"} return "gallerydl" else: match keyname.lower(): case "youtube": - return "ytdl" + return {"tool": "ytdl", "opts": normopts} case "tiktok": - return "ytdl" + return {"tool": "ytdl", "opts": tiktokopts} case "website": - return "website" + return {"tool": "website"} + case "websites": + return {"tool": "website"} case "site": - return "website" - return "gallerydl" - -def videohost(sitename: str, urls: list[str]): - opts = {'allsubtitles': True, - 'extract_flat': 'discard_in_playlist', - 'format': 'bestvideo[height>=720][fps>=60]+bestaudio/bestvideo+bestaudio/best', - 'fragment_retries': 10, - 'ignoreerrors': True, - 'outtmpl': {'default': sitename + '/%(uploader|author)s/%(title)s.%(ext)s'}, - 'postprocessors': [{'key': 'FFmpegConcat', - 'only_multi_video': True, - 'when': 'playlist'}], - 'retries': 10, - 'writeannotations': True, - 'writedescription': True, - 'writeinfojson': True, - 'writesubtitles': True, - 'writethumbnail': True} + return {"tool": "website"} + return {"tool": "gallerydl"} +def videohost(sitename: str, urls: list[str], opts): with yt_dlp.YoutubeDL(opts) as ydl: ydl.download(urls) @@ -98,14 +127,16 @@ def archivesite(urls: list[str]): for key in input["sites"]: print("Downloading for " + key) - program = figureUse(input["sites"][key], key) + programData = figureUse(input["sites"][key], key) - print("Opting to use: " + program) - match program: + print("Opting to use: " + programData["tool"]) + match programData["tool"]: case "ytdl": - videohost(key, input["sites"][key]["urls"]) + videohost(key, input["sites"][key]["urls"], programData["opts"]) case "gallerydl": gallery(input["sites"][key]["urls"]) case "website": archivesite(input["sites"][key]["urls"]) + case "websites": + archivesite(input["sites"][key]["urls"]) |