mirror of
https://github.com/karl0ss/GoGoDownloader.git
synced 2025-04-26 11:39:22 +01:00
74 lines
2.1 KiB
Python
74 lines
2.1 KiB
Python
|
from backend import *
|
||
|
|
import argparse
|
||
|
|
import os
|
||
|
|
import requests
|
||
|
|
from bs4 import BeautifulSoup
|
||
|
|
|
||
|
|
|
||
|
|
def get_archives(archive_file):
|
||
|
|
|
||
|
|
if os.path.isfile(archive_file):
|
||
|
|
with open(archive_file, "r") as f:
|
||
|
|
return f.read().split()
|
||
|
|
|
||
|
|
else:
|
||
|
|
return []
|
||
|
|
|
||
|
|
|
||
|
|
def find_episodes(url):
|
||
|
|
resp = requests.get(url)
|
||
|
|
soup = BeautifulSoup(resp.text, "html.parser")
|
||
|
|
all_episodes = soup.find("ul", {"id": "episode_page"})
|
||
|
|
all_episodes = int(all_episodes.get_text().split("-")[-1].strip())
|
||
|
|
return all_episodes
|
||
|
|
|
||
|
|
|
||
|
|
def main():
|
||
|
|
placeholder = "DUMMY"
|
||
|
|
config = config_check()
|
||
|
|
parser = argparse.ArgumentParser()
|
||
|
|
parser.add_argument("url", metavar="url", type=str)
|
||
|
|
parser.add_argument("--archive", metavar="Archive File", type=str)
|
||
|
|
parser.add_argument(
|
||
|
|
"--quality",
|
||
|
|
metavar="download quality",
|
||
|
|
choices={"360", "480", "720", "1080"},
|
||
|
|
nargs="?",
|
||
|
|
const="1080",
|
||
|
|
default="1080",
|
||
|
|
type=str,
|
||
|
|
)
|
||
|
|
args = parser.parse_args()
|
||
|
|
|
||
|
|
name = args.url.split("/")[-1]
|
||
|
|
title = name.replace("-", " ").title().strip()
|
||
|
|
all_episodes = find_episodes(args.url)
|
||
|
|
|
||
|
|
archives = get_archives(args.archive)
|
||
|
|
downloader = gogoanime(
|
||
|
|
config, name, args.quality, placeholder, all_episodes, 1, all_episodes, title
|
||
|
|
)
|
||
|
|
|
||
|
|
episode_links = [link for link in downloader.get_links() if link not in archives]
|
||
|
|
dl_links = {}
|
||
|
|
for ep in episode_links:
|
||
|
|
ep_num = ep.split("-")[-1]
|
||
|
|
dl_links[downloader.get_download_link(ep)] = (title, ep_num)
|
||
|
|
results = downloader.file_downloader(dl_links)
|
||
|
|
failed = []
|
||
|
|
with open(args.archive, "a+") as f:
|
||
|
|
for ep in episode_links:
|
||
|
|
ep_num = ep.split("-")[-1]
|
||
|
|
if os.path.join(title, f"{title} Episode {ep_num}.mp4") in results:
|
||
|
|
f.write(f"{ep}\n")
|
||
|
|
else:
|
||
|
|
failed.append(ep)
|
||
|
|
|
||
|
|
with open("logs.txt", "w+") as f:
|
||
|
|
for failure in failed:
|
||
|
|
f.write(f"{failed} failed to download\n")
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
main()
|