From 0c7115dda24d271029d4dbeb5255b563d776e096 Mon Sep 17 00:00:00 2001
From: "Sp4rk.y" <sp4rky@cdm-project.com>
Date: Sat, 7 Sep 2024 20:33:40 -0600
Subject: [PATCH] =?UTF-8?q?=E2=99=BB=EF=B8=8F=20(DROP/=5F=5Finit=5F=5F.py)?=
 =?UTF-8?q?:=20reformat=20long=20lines=20for=20better=20readability=20and?=
 =?UTF-8?q?=20maintainability?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 services/DROP/__init__.py | 35 +++++++++++++++++++----------------
 1 file changed, 19 insertions(+), 16 deletions(-)

diff --git a/services/DROP/__init__.py b/services/DROP/__init__.py
index a518fc1..c278e16 100644
--- a/services/DROP/__init__.py
+++ b/services/DROP/__init__.py
@@ -26,7 +26,9 @@ class DROP(Service):
 
     TITLE_RE = r"^(?:https?://(?:www\.)?dropout\.tv/)([^/]+)(?:/.*)?$"
     SERIES_RE = r"https?://(?:www\.)?dropout\.tv/([^/]+)(?:/season:(\d+))?/?$"
-    EPISODE_RE = r"https?://(?:www\.)?dropout\.tv/([^/]+)/season:(\d+)/videos/([^/]+)/?$"
+    EPISODE_RE = (
+        r"https?://(?:www\.)?dropout\.tv/([^/]+)/season:(\d+)/videos/([^/]+)/?$"
+    )
 
     LOGIN_URL = "https://www.dropout.tv/login"
 
@@ -58,7 +60,9 @@ class DROP(Service):
                 "utf8": "true",
             }
 
-            response = self.session.post(self.LOGIN_URL, data=login_data, allow_redirects=False)
+            response = self.session.post(
+                self.LOGIN_URL, data=login_data, allow_redirects=False
+            )
 
             if '<div id="watch-unauthorized"' in response.text:
                 self.log.error("Login failed")
@@ -66,7 +70,9 @@ class DROP(Service):
             else:
                 self.log.info("Login successful")
         else:
-            self.log.info("No login credentials provided, proceeding without authentication")
+            self.log.info(
+                "No login credentials provided, proceeding without authentication"
+            )
 
     def _get_authenticity_token(self):
         signin_page = self.session.get(self.LOGIN_URL).text
@@ -106,14 +112,20 @@ class DROP(Service):
                 episode_link = item.find("a", class_="browse-item-link")
                 if episode_link:
                     episode_url = episode_link["href"]
-                    episode_data = json.loads(episode_link["data-track-event-properties"])
+                    episode_data = json.loads(
+                        episode_link["data-track-event-properties"]
+                    )
 
                     episode_id = episode_data["id"]
                     episode_title = episode_data["label"]
 
-                    episode_number_elem = item.find("span", class_="media-identifier media-episode")
+                    episode_number_elem = item.find(
+                        "span", class_="media-identifier media-episode"
+                    )
                     if episode_number_elem:
-                        episode_number_match = re.search(r"Episode (\d+)", episode_number_elem.text)
+                        episode_number_match = re.search(
+                            r"Episode (\d+)", episode_number_elem.text
+                        )
                         if episode_number_match:
                             episode_number = int(episode_number_match.group(1))
                         else:
@@ -169,21 +181,13 @@ class DROP(Service):
         cdn = cdns.get(default_cdn) or next(iter(cdns.values()))
         mpd_url = cdn["avc_url"].replace("playlist.json", "playlist.mpd")
         tracks = HLS.from_url(url=mpd_url).to_tracks(language="en")
-
-        # Extract thumbnail URL from config_data
         thumbnail_base_url = config_data["video"]["thumbs"]["base"]
         thumbnail_url = f"{thumbnail_base_url}"
-
-        # Download the thumbnail
         thumbnail_response = self.session.get(thumbnail_url)
         if thumbnail_response.status_code == 200:
             thumbnail_filename = f"{title.id}_thumbnail.jpg"
             thumbnail_path = config.directories.temp / thumbnail_filename
-
-            # Ensure the directory exists
             os.makedirs(config.directories.temp, exist_ok=True)
-
-            # Save the thumbnail file
             with open(thumbnail_path, "wb") as f:
                 f.write(thumbnail_response.content)
 
@@ -192,10 +196,9 @@ class DROP(Service):
                 path=thumbnail_path,
                 name=thumbnail_filename,
                 mime_type="image/jpeg",
-                description="Thumbnail"
+                description="Thumbnail",
             )
 
-            # Add the attachment to the tracks
             tracks.attachments.append(thumbnail_attachment)
 
         return tracks