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

Update backend.py

Browse Source 
Small 3kb files results from download links with no files. link with no files gives out status code 200. built in url check when checking for resolution
This commit is contained in:
Arctic4161 2021-10-14 21:57:41 -05:00 committed by GitHub
parent d1348cbd95
commit e9e89e3159
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 20 additions and 43 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

63
src/backend.py
View File

@ -58,19 +58,31 @@ class Download:
def get_download_urls(self, download_link):
episode_quality = self.episode_quality
if episode_quality == "FullHDP":
episode_quality = "1080P"
episode_quality = "1080P - mp4"
elif episode_quality == "HDP":
episode_quality = "720P"
episode_quality = "720P - mp4"
elif episode_quality == "SDP":
episode_quality = "360P"
episode_quality = "360P - mp4"
else:
episode_quality = "1080P"
with req.get(download_link) as res:
soup = BeautifulSoup(res.content, "html.parser")
link = soup.find("div", {"class": "dowload"}, text=re.compile(episode_quality))
if link is None:
pass
else:
with req.get(link.a.get("href")) as checkingurl:
if checkingurl.status_code == 200:
link = None
if link is None:
episode_quality = "720P"
link = soup.find("div", {"class": "dowload"}, text=re.compile(episode_quality))
if link is None:
pass
else:
with req.get(link.a.get("href")) as checkingurl:
if checkingurl.status_code == 200:
link = None
if link is None:
episode_quality = "360P"
link = soup.find("div", {"class": "dowload"}, text=re.compile(episode_quality))
@ -100,45 +112,11 @@ class Download:
def download_episodes(self, url):
client = req.session()
while True:
with client.get(url[1], headers=self.random_headers(), stream=True, timeout=10) as workingurl:
episode_name = "EP." + url[0] + ".mp4"
file_loc = os.path.join(self.folder, episode_name)
with open(file_loc, "w+b") as file:
shutil.copyfileobj(workingurl.raw, file)
size = os.stat(file_loc).st_size
if int(size) < 5:
with client.get(url[1], headers=self.random_headers(), stream=True, timeout=10) as workingurl:
episode_name = "EP." + url[0] + ".mp4"
file_loc = os.path.join(self.folder, episode_name)
with open(file_loc, "w+b") as file:
shutil.copyfileobj(workingurl.raw, file)
size = os.stat(file_loc).st_size
if int(size) < 5:
with client.get(url[1], headers=self.random_headers(), stream=True, timeout=10) as workingurl:
episode_name = "EP." + url[0] + ".mp4"
file_loc = os.path.join(self.folder, episode_name)
with open(file_loc, "w+b") as file:
shutil.copyfileobj(workingurl.raw, file)
size = os.stat(file_loc).st_size
if int(size) < 5:
with client.get(url[1], headers=self.random_headers(), stream=True, timeout=10) as workingurl:
episode_name = "EP." + url[0] + ".mp4"
file_loc = os.path.join(self.folder, episode_name)
with open(file_loc, "w+b") as file:
shutil.copyfileobj(workingurl.raw, file)
size = os.stat(file_loc).st_size
if int(size) < 5:
with client.get(url[1], headers=self.random_headers(), stream=True, timeout=10) as workingurl:
episode_name = "EP." + url[0] + ".mp4"
file_loc = os.path.join(self.folder, episode_name)
with open(file_loc, "w+b") as file:
shutil.copyfileobj(workingurl.raw, file)
break
with client.get(url[1], headers=self.random_headers(), stream=True, timeout=10) as workingurl:
episode_name = "EP." + url[0] + ".mp4"
file_loc = os.path.join(self.folder, episode_name)
with open(file_loc, "w+b") as file:
shutil.copyfileobj(workingurl.raw, file, 8192)
@dataclass(init=True)
@ -160,4 +138,3 @@ class CustomMessage(Exception):
print(
f"{Fore.RESET}[{Fore.GREEN}+{Fore.RESET}] Using {Fore.LIGHTCYAN_EX}{self.episode_quality}{Fore.RESET} as a default quality."
)