Karl/GoGoDownloader
1
0
Fork 0
mirror of https://github.com/karl0ss/GoGoDownloader.git synced 2025-04-26 19:49:23 +01:00
Code Issues Packages Projects Releases Wiki Activity

Refactored

Browse Source
This commit is contained in:
sh1nobu 2021-09-22 17:46:42 +08:00
parent 6b59251753
commit 218eb40c40
1 changed files with 34 additions and 47 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

81
src/bitanime.py
View File

@ -3,14 +3,15 @@ import ctypes
import os import os
import colorama import colorama
import concurrent.futures import concurrent.futures
from backend import Download, CustomError from backend import Download, CustomMessage
from tqdm.contrib.concurrent import thread_map from tqdm.contrib.concurrent import thread_map
from bs4 import BeautifulSoup from bs4 import BeautifulSoup
from colorama import Fore from colorama import Fore
colorama.init(autoreset=True) colorama.init(autoreset=True)
PLUS = f"[{Fore.GREEN}+{Fore.RESET}] " OK = f"[{Fore.GREEN}+{Fore.RESET}] "
MINUS = f"[{Fore.RED}-{Fore.RESET}] {Fore.LIGHTRED_EX}" ERR = f"[{Fore.RED}-{Fore.RESET}] "
IN = f"[{Fore.LIGHTBLUE_EX}>{Fore.RESET}] "
try: try:
ctypes.windll.kernel32.SetConsoleTitleW("BitAnime") ctypes.windll.kernel32.SetConsoleTitleW("BitAnime")
except (AttributeError): except (AttributeError):
@ -32,7 +33,7 @@ def bitanime():
""" """
) )
while True: while True:
name = input(f"{PLUS}Enter anime name > ").lower() name = input(f"{IN}Enter anime name > ").lower()
if "-" in name: if "-" in name:
title = name.replace("-", " ").title().strip() title = name.replace("-", " ").title().strip()
else: else:
@ -45,10 +46,10 @@ def bitanime():
all_episodes = int(all_episodes.get_text().split("-")[-1].strip()) all_episodes = int(all_episodes.get_text().split("-")[-1].strip())
break break
else: else:
print(f"{MINUS}Error 404: Anime not found. Please try again.") print(f"{ERR}Error 404: Anime not found. Please try again.")
while True: while True:
quality = input( quality = input(
f"{PLUS}Enter episode quality (1.SD/360P|2.HD/720P|3.FULLHD/1080P) > " f"{IN}Enter episode quality (1.SD/360P|2.HD/720P|3.FULLHD/1080P) > "
) )
if quality == "1" or quality == "": if quality == "1" or quality == "":
episode_quality = "SDP" episode_quality = "SDP"
@ -60,11 +61,11 @@ def bitanime():
episode_quality = "FullHDP" episode_quality = "FullHDP"
break break
else: else:
print(f"{MINUS}Invalid input. Please try again.") print(f"{ERR}Invalid input. Please try again.")
print(f"{PLUS}Title: {Fore.LIGHTCYAN_EX}{title}") print(f"{OK}Title: {Fore.LIGHTCYAN_EX}{title}")
print(f"{PLUS}Episode/s: {Fore.LIGHTCYAN_EX}{all_episodes}") print(f"{OK}Episode/s: {Fore.LIGHTCYAN_EX}{all_episodes}")
print(f"{PLUS}Quality: {Fore.LIGHTCYAN_EX}{episode_quality}") print(f"{OK}Quality: {Fore.LIGHTCYAN_EX}{episode_quality}")
print(f"{PLUS}Link: {Fore.LIGHTCYAN_EX}{source}") print(f"{OK}Link: {Fore.LIGHTCYAN_EX}{source}")
folder = os.path.join(os.getcwd(), title) folder = os.path.join(os.getcwd(), title)
if not os.path.exists(folder): if not os.path.exists(folder):
os.mkdir(folder) os.mkdir(folder)
@ -74,12 +75,12 @@ def bitanime():
if all_episodes != 1: if all_episodes != 1:
while True: while True:
choice = input( choice = input(
f"{PLUS}Do you want to download all episode? (y/n) > " f"{IN}Do you want to download all episode? (y/n) > "
).lower() ).lower()
if choice in ["y", "n"]: if choice in ["y", "n"]:
break break
else: else:
print(f"{MINUS}Invalid input. Please try again.") print(f"{ERR}Invalid input. Please try again.")
episode_start = None episode_start = None
episode_end = None episode_end = None
@ -87,24 +88,24 @@ def bitanime():
if choice == "n": if choice == "n":
while True: while True:
try: try:
episode_start = int(input(f"{PLUS}Episode start > ")) episode_start = int(input(f"{IN}Episode start > "))
episode_end = int(input(f"{PLUS}Episode end > ")) episode_end = int(input(f"{IN}Episode end > "))
if episode_start <= 0 or episode_end <= 0: if episode_start <= 0 or episode_end <= 0:
CustomError( CustomMessage(
f"{MINUS}episode_start or episode_end cannot be less than or equal to 0" f"{ERR}episode_start or episode_end cannot be less than or equal to 0"
).print_error() ).print_error()
elif episode_start >= all_episodes or episode_end > all_episodes: elif episode_start >= all_episodes or episode_end > all_episodes:
CustomError( CustomMessage(
f"{MINUS}episode_start or episode_end cannot be more than {all_episodes}" f"{ERR}episode_start or episode_end cannot be more than {all_episodes}"
).print_error() ).print_error()
elif episode_end <= episode_start: elif episode_end <= episode_start:
CustomError( CustomMessage(
f"{MINUS}episode_end cannot be less than or equal to episode_start" f"{ERR}episode_end cannot be less than or equal to episode_start"
).print_error() ).print_error()
else: else:
break break
except ValueError: except ValueError:
print(f"{MINUS}Invalid input. Please try again.") print(f"{ERR}Invalid input. Please try again.")
episode_start = episode_start if episode_start is not None else 1 episode_start = episode_start if episode_start is not None else 1
episode_end = episode_end if episode_end is not None else all_episodes episode_end = episode_end if episode_end is not None else all_episodes
@ -116,31 +117,17 @@ def bitanime():
source = f"https://gogoanime.pe/{name}" source = f"https://gogoanime.pe/{name}"
with req.get(source) as res: with req.get(source) as res:
soup = BeautifulSoup(res.content, "html.parser") soup = BeautifulSoup(res.content, "html.parser")
episode_zero = soup.find("h1", {"class": "entry-title"}) episode_zero = soup.find("h1", {"class": "entry-title"}) # value: 404
if episode_zero is None:
# Episode 0 == 200 if choice == "n" or episode_zero is not None:
with concurrent.futures.ThreadPoolExecutor() as exec: source = None
episode_links = download.get_links(
source=source if choice == "y" else None episode_links = download.get_links(source)
) with concurrent.futures.ThreadPoolExecutor() as exec:
download_links = list( download_links = list(exec.map(download.get_download_links, episode_links))
exec.map(download.get_download_links, episode_links) download_urls = list(exec.map(download.get_download_urls, download_links))
)
download_urls = list(
exec.map(download.get_download_urls, download_links)
)
else:
# Episode 0 == 404
with concurrent.futures.ThreadPoolExecutor() as exec:
episode_links = download.get_links()
download_links = list(
exec.map(download.get_download_links, episode_links)
)
download_urls = list(
exec.map(download.get_download_urls, download_links)
)
print( print(
f"{PLUS}Downloading {Fore.LIGHTCYAN_EX}{len(download_urls)}{Fore.RESET} episode/s" f"{OK}Downloading {Fore.LIGHTCYAN_EX}{len(download_urls)}{Fore.RESET} episode/s"
) )
thread_map( thread_map(
download.download_episodes, download.download_episodes,
@ -155,7 +142,7 @@ def bitanime():
opener = "open" if sys.platform == "darwin" else "xdg-open" opener = "open" if sys.platform == "darwin" else "xdg-open"
subprocess.call([opener, folder]) subprocess.call([opener, folder])
use_again = input(f"{PLUS}Do you want to use the app again? (y|n) > ").lower() use_again = input(f"{IN}Do you want to use the app again? (y|n) > ").lower()
if use_again == "y": if use_again == "y":
os.system("cls") os.system("cls")
else: else: