From 2df3c5ce1ae4552b2388f1b92804d38028b3fa05 Mon Sep 17 00:00:00 2001
From: "karl.hudgell" <karl.hudgell@bjss.com>
Date: Wed, 19 Jan 2022 10:16:34 +0000
Subject: [PATCH] move everything to download class

---
 backend.py | 76 ++++++++++++++++++++++++++++--------------------------
 main.py    |  9 ++++---
 2 files changed, 44 insertions(+), 41 deletions(-)

diff --git a/backend.py b/backend.py
index c4d1935..4cf4684 100644
--- a/backend.py
+++ b/backend.py
@@ -36,13 +36,14 @@ CURRENT_DOMAIN = "film"
 
 @dataclass(init=True)
 class Download:
+    config: object
     name: str
     episode_quality: str
     folder: str
     all_episodes: int
     episode_start: int
     episode_end: int
-    config: object
+    title: str
     printed: bool = False
 
     def get_links(self, source=None):
@@ -61,45 +62,46 @@ class Download:
             ]
         return episode_links
 
+    def get_download_link(self, url):
 
-def get_download_link(config, url, episode_quality):
-
-    page = requests.get(
-        url,
-        cookies=dict(auth=config["GoGoAnimeAuthKey"]),
-    )
-
-    soup = BeautifulSoup(page.content, "html.parser")
-
-    for link in soup.find_all("a", href=True):
-        if episode_quality in link.text:
-            return link["href"]
-
-
-def file_downloader(file_list: dict, title: str, config: object):
-    dl = Downloader(
-        max_conn=config["MaxConcurrentDownloads"],
-        overwrite=False,
-        headers=dict(
-            [
-                (
-                    "User-Agent",
-                    "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/97.0.4692.71 Safari/537.36",
-                ),
-                ("authority", "gogo-cdn.com"),
-                ("referer", f"https://gogoanime.{config['CurrentGoGoAnimeDomain']}/"),
-            ]
-        ),
-    )
-
-    for link in file_list:
-        dl.enqueue_file(
-            link,
-            path=f"./{title}",
+        page = requests.get(
+            url,
+            cookies=dict(auth=self.config["GoGoAnimeAuthKey"]),
         )
 
-    files = dl.download()
-    return files
+        soup = BeautifulSoup(page.content, "html.parser")
+
+        for link in soup.find_all("a", href=True):
+            if self.episode_quality in link.text:
+                return link["href"]
+
+    def file_downloader(self, file_list: dict):
+        dl = Downloader(
+            max_conn=self.config["MaxConcurrentDownloads"],
+            overwrite=self.config["OverwriteDownloads"],
+            headers=dict(
+                [
+                    (
+                        "User-Agent",
+                        "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/97.0.4692.71 Safari/537.36",
+                    ),
+                    ("authority", "gogo-cdn.com"),
+                    (
+                        "referer",
+                        f"https://gogoanime.{self.config['CurrentGoGoAnimeDomain']}/",
+                    ),
+                ]
+            ),
+        )
+
+        for link in file_list:
+            dl.enqueue_file(
+                link,
+                path=f"./{self.title}",
+            )
+
+        files = dl.download()
+        return files
 
 
 @dataclass(init=True)
diff --git a/main.py b/main.py
index 6747a80..6f95741 100644
--- a/main.py
+++ b/main.py
@@ -1,7 +1,7 @@
 import requests
 import ctypes
 import os
-from backend import *
+from backend import Download, CustomMessage, config_check
 from bs4 import BeautifulSoup
 from colorama import Fore
 
@@ -126,13 +126,14 @@ def gogodownloader(config):
             episode_end = all_episodes
 
         download = Download(
+            config,
             name,
             episode_quality,
             folder,
             all_episodes,
             episode_start,
             episode_end,
-            config,
+            title,
         )
 
         source = f"https://gogoanime.{CURRENT_DOMAIN}/{name}"
@@ -147,9 +148,9 @@ def gogodownloader(config):
         episode_links = download.get_links(source)
 
         for link in episode_links:
-            dl_links.append(get_download_link(config, link, episode_quality))
+            dl_links.append(download.get_download_link(link))
 
-        file_downloader(dl_links, title, config)
+        download.file_downloader(dl_links)
 
         use_again = input(f"{IN}Do you want to use the app again? (y|n) > ").lower()
         if use_again == "y":