quasarr 2.4.8__py3-none-any.whl → 2.4.9__py3-none-any.whl

quasarr/search/sources/mb.py

@@ -12,9 +12,9 @@ from urllib.parse import quote_plus
 import requests
 from bs4 import BeautifulSoup
 
-from quasarr.providers.hostname_issues import mark_hostname_issue, clear_hostname_issue
+from quasarr.providers.hostname_issues import clear_hostname_issue, mark_hostname_issue
 from quasarr.providers.imdb_metadata import get_localized_title
-from quasarr.providers.log import info, debug
+from quasarr.providers.log import debug, info
 
 hostname = "mb"
 supported_mirrors = ["rapidgator", "ddownload"]
@@ -25,8 +25,18 @@ IMDB_REGEX = re.compile(r"imdb\.com/title/(tt\d+)")
 
 # map German month names to numbers
 GERMAN_MONTHS = {
-    'Januar': '01', 'Februar': '02', 'März': '03', 'April': '04', 'Mai': '05', 'Juni': '06',
-    'Juli': '07', 'August': '08', 'September': '09', 'Oktober': '10', 'November': '11', 'Dezember': '12'
+    "Januar": "01",
+    "Februar": "02",
+    "März": "03",
+    "April": "04",
+    "Mai": "05",
+    "Juni": "06",
+    "Juli": "07",
+    "August": "08",
+    "September": "09",
+    "Oktober": "10",
+    "November": "11",
+    "Dezember": "12",
 }
 
 
@@ -42,11 +52,19 @@ def extract_size(text):
     return {"size": m.group(1), "sizeunit": m.group(2)}
 
 
-def _parse_posts(soup, shared_state, password, mirror_filter,
-                 is_search=False, request_from=None, search_string=None,
-                 season=None, episode=None):
+def _parse_posts(
+    soup,
+    shared_state,
+    password,
+    mirror_filter,
+    is_search=False,
+    request_from=None,
+    search_string=None,
+    season=None,
+    episode=None,
+):
     releases = []
-    one_hour_ago = (datetime.now() - timedelta(hours=1)).strftime('%Y-%m-%d %H:%M:%S')
+    one_hour_ago = (datetime.now() - timedelta(hours=1)).strftime("%Y-%m-%d %H:%M:%S")
 
     for post in soup.select("div.post"):
         try:
@@ -61,23 +79,25 @@ def _parse_posts(soup, shared_state, password, mirror_filter,
             date_txt = date_p.get_text(strip=True) if date_p else None
             published = one_hour_ago
             if date_txt:
-                m_date = re.search(r'(?:\w+, )?(\d{1,2})\.\s*(\w+)\s+(\d{4})\s+(\d{2}:\d{2})', date_txt)
+                m_date = re.search(
+                    r"(?:\w+, )?(\d{1,2})\.\s*(\w+)\s+(\d{4})\s+(\d{2}:\d{2})", date_txt
+                )
                 if m_date:
                     day, mon_name, year, hm = m_date.groups()
-                    mon = GERMAN_MONTHS.get(mon_name, '01')
-                    dt_obj = datetime.strptime(f"{day}.{mon}.{year} {hm}", "%d.%m.%Y %H:%M")
+                    mon = GERMAN_MONTHS.get(mon_name, "01")
+                    dt_obj = datetime.strptime(
+                        f"{day}.{mon}.{year} {hm}", "%d.%m.%Y %H:%M"
+                    )
                     published = dt_obj.strftime("%a, %d %b %Y %H:%M:%S +0000")
 
             if is_search:
-                if not shared_state.is_valid_release(title,
-                                                     request_from,
-                                                     search_string,
-                                                     season,
-                                                     episode):
+                if not shared_state.is_valid_release(
+                    title, request_from, search_string, season, episode
+                ):
                     continue
 
                 # drop .XXX. unless user explicitly searched xxx
-                if XXX_REGEX.search(title) and 'xxx' not in search_string.lower():
+                if XXX_REGEX.search(title) and "xxx" not in search_string.lower():
                     continue
                 # require resolution/codec
                 if not (RESOLUTION_REGEX.search(title) or CODEC_REGEX.search(title)):
@@ -91,10 +111,12 @@ def _parse_posts(soup, shared_state, password, mirror_filter,
                     continue
             else:
                 mirror_candidates = []
-                for strong in post.find_all('strong', string=re.compile(r'^Download', re.I)):
-                    link_tag = strong.find_next_sibling('a')
+                for strong in post.find_all(
+                    "strong", string=re.compile(r"^Download", re.I)
+                ):
+                    link_tag = strong.find_next_sibling("a")
                     if link_tag and link_tag.get_text(strip=True):
-                        host = link_tag.get_text(strip=True).split('.')[0].lower()
+                        host = link_tag.get_text(strip=True).split(".")[0].lower()
                         mirror_candidates.append(host)
                 valid = [m for m in mirror_candidates if m in supported_mirrors]
                 if not valid or (mirror_filter and mirror_filter not in valid):
@@ -102,8 +124,8 @@ def _parse_posts(soup, shared_state, password, mirror_filter,
 
             # extract IMDb ID
             imdb_id = None
-            for tag in post.find_all('a', href=True):
-                m = IMDB_REGEX.search(tag['href'])
+            for tag in post.find_all("a", href=True):
+                m = IMDB_REGEX.search(tag["href"])
                 if m:
                     imdb_id = m.group(1)
                     break
@@ -119,21 +141,25 @@ def _parse_posts(soup, shared_state, password, mirror_filter,
             payload = urlsafe_b64encode(
                 f"{title}|{source}|{mirror_filter}|{mb}|{password}|{imdb_id}|{hostname}".encode()
             ).decode()
-            link = f"{shared_state.values['internal_address']}/download/?payload={payload}"
-
-            releases.append({
-                "details": {
-                    "title": title,
-                    "hostname": hostname,
-                    "imdb_id": imdb_id,
-                    "link": link,
-                    "mirror": mirror_filter,
-                    "size": size_bytes,
-                    "date": published,
-                    "source": source
-                },
-                "type": "protected"
-            })
+            link = (
+                f"{shared_state.values['internal_address']}/download/?payload={payload}"
+            )
+
+            releases.append(
+                {
+                    "details": {
+                        "title": title,
+                        "hostname": hostname,
+                        "imdb_id": imdb_id,
+                        "link": link,
+                        "mirror": mirror_filter,
+                        "size": size_bytes,
+                        "date": published,
+                        "source": source,
+                    },
+                    "type": "protected",
+                }
+            )
         except Exception as e:
             debug(f"Error parsing {hostname.upper()} post: {e}")
             continue
@@ -144,13 +170,15 @@ def mb_feed(shared_state, start_time, request_from, mirror=None):
     mb = shared_state.values["config"]("Hostnames").get(hostname)
 
     if not "arr" in request_from.lower():
-        debug(f'Skipping {request_from} search on "{hostname.upper()}" (unsupported media type)!')
+        debug(
+            f'Skipping {request_from} search on "{hostname.upper()}" (unsupported media type)!'
+        )
         return []
 
     password = mb
     section = "neuerscheinungen" if "Radarr" in request_from else "serie"
     url = f"https://{mb}/category/{section}/"
-    headers = {'User-Agent': shared_state.values["user_agent"]}
+    headers = {"User-Agent": shared_state.values["user_agent"]}
     try:
         r = requests.get(url, headers=headers, timeout=30)
         r.raise_for_status()
@@ -158,7 +186,9 @@ def mb_feed(shared_state, start_time, request_from, mirror=None):
         releases = _parse_posts(soup, shared_state, password, mirror_filter=mirror)
     except Exception as e:
         info(f"Error loading {hostname.upper()} feed: {e}")
-        mark_hostname_issue(hostname, "feed", str(e) if "e" in dir() else "Error occurred")
+        mark_hostname_issue(
+            hostname, "feed", str(e) if "e" in dir() else "Error occurred"
+        )
         releases = []
     debug(f"Time taken: {time.time() - start_time:.2f}s ({hostname})")
 
@@ -167,17 +197,27 @@ def mb_feed(shared_state, start_time, request_from, mirror=None):
     return releases
 
 
-def mb_search(shared_state, start_time, request_from, search_string, mirror=None, season=None, episode=None):
+def mb_search(
+    shared_state,
+    start_time,
+    request_from,
+    search_string,
+    mirror=None,
+    season=None,
+    episode=None,
+):
     mb = shared_state.values["config"]("Hostnames").get(hostname)
 
     if not "arr" in request_from.lower():
-        debug(f'Skipping {request_from} search on "{hostname.upper()}" (unsupported media type)!')
+        debug(
+            f'Skipping {request_from} search on "{hostname.upper()}" (unsupported media type)!'
+        )
         return []
 
     password = mb
     imdb_id = shared_state.is_imdb_id(search_string)
     if imdb_id:
-        title = get_localized_title(shared_state, imdb_id, 'de')
+        title = get_localized_title(shared_state, imdb_id, "de")
         if not title:
             info(f"Could not extract title from IMDb-ID {imdb_id}")
             return []
@@ -185,19 +225,27 @@ def mb_search(shared_state, start_time, request_from, search_string, mirror=None
 
     q = quote_plus(search_string)
     url = f"https://{mb}/?s={q}&id=20&post_type=post"
-    headers = {'User-Agent': shared_state.values["user_agent"]}
+    headers = {"User-Agent": shared_state.values["user_agent"]}
     try:
         r = requests.get(url, headers=headers, timeout=10)
         r.raise_for_status()
         soup = BeautifulSoup(r.content, "html.parser")
         releases = _parse_posts(
-            soup, shared_state, password, mirror_filter=mirror,
-            is_search=True, request_from=request_from,
-            search_string=search_string, season=season, episode=episode
+            soup,
+            shared_state,
+            password,
+            mirror_filter=mirror,
+            is_search=True,
+            request_from=request_from,
+            search_string=search_string,
+            season=season,
+            episode=episode,
         )
     except Exception as e:
         info(f"Error loading {hostname.upper()} search: {e}")
-        mark_hostname_issue(hostname, "search", str(e) if "e" in dir() else "Error occurred")
+        mark_hostname_issue(
+            hostname, "search", str(e) if "e" in dir() else "Error occurred"
+        )
         releases = []
     debug(f"Time taken: {time.time() - start_time:.2f}s ({hostname})")
 

quasarr/search/sources/nk.py

@@ -12,9 +12,9 @@ from urllib.parse import urljoin
 import requests
 from bs4 import BeautifulSoup
 
-from quasarr.providers.hostname_issues import mark_hostname_issue, clear_hostname_issue
+from quasarr.providers.hostname_issues import clear_hostname_issue, mark_hostname_issue
 from quasarr.providers.imdb_metadata import get_localized_title, get_year
-from quasarr.providers.log import info, debug
+from quasarr.providers.log import debug, info
 
 hostname = "nk"
 supported_mirrors = ["rapidgator", "ddownload"]
@@ -22,7 +22,12 @@ supported_mirrors = ["rapidgator", "ddownload"]
 
 def convert_to_rss_date(date_str: str) -> str:
     date_str = date_str.strip()
-    for fmt in ("%d. %B %Y / %H:%M", "%d.%m.%Y / %H:%M", "%d.%m.%Y - %H:%M", "%Y-%m-%d %H:%M"):
+    for fmt in (
+        "%d. %B %Y / %H:%M",
+        "%d.%m.%Y / %H:%M",
+        "%d.%m.%Y - %H:%M",
+        "%Y-%m-%d %H:%M",
+    ):
         try:
             dt = datetime.strptime(date_str, fmt)
             return dt.strftime("%a, %d %b %Y %H:%M:%S +0000")
@@ -34,33 +39,43 @@ def convert_to_rss_date(date_str: str) -> str:
 def extract_size(text: str) -> dict:
     match = re.search(r"(\d+(?:[\.,]\d+)?)\s*([A-Za-z]+)", text)
     if match:
-        size = match.group(1).replace(',', '.')
+        size = match.group(1).replace(",", ".")
         unit = match.group(2)
         return {"size": size, "sizeunit": unit}
     return {"size": "0", "sizeunit": "MB"}
 
 
 def get_release_field(res, label):
-    for li in res.select('ul.release-infos li'):
-        sp = li.find('span')
+    for li in res.select("ul.release-infos li"):
+        sp = li.find("span")
         if not sp:
-            return ''
+            return ""
         if sp.get_text(strip=True).lower() == label.lower():
-            txt = li.get_text(' ', strip=True)
-            return txt[len(sp.get_text(strip=True)):].strip()
-    return ''
+            txt = li.get_text(" ", strip=True)
+            return txt[len(sp.get_text(strip=True)) :].strip()
+    return ""
 
 
 def nk_feed(*args, **kwargs):
     return nk_search(*args, **kwargs)
 
 
-def nk_search(shared_state, start_time, request_from, search_string="", mirror=None, season=None, episode=None):
+def nk_search(
+    shared_state,
+    start_time,
+    request_from,
+    search_string="",
+    mirror=None,
+    season=None,
+    episode=None,
+):
     releases = []
     host = shared_state.values["config"]("Hostnames").get(hostname)
 
     if not "arr" in request_from.lower():
-        debug(f'Skipping {request_from} search on "{hostname.upper()}" (unsupported media type)!')
+        debug(
+            f'Skipping {request_from} search on "{hostname.upper()}" (unsupported media type)!'
+        )
         return releases
 
     if mirror and mirror not in supported_mirrors:
@@ -71,7 +86,7 @@ def nk_search(shared_state, start_time, request_from, search_string="", mirror=N
     if search_string != "":
         imdb_id = shared_state.is_imdb_id(search_string)
         if imdb_id:
-            local_title = get_localized_title(shared_state, imdb_id, 'de')
+            local_title = get_localized_title(shared_state, imdb_id, "de")
             if not local_title:
                 info(f"{hostname}: no title for IMDb {imdb_id}")
                 return releases
@@ -99,18 +114,20 @@ def nk_search(shared_state, start_time, request_from, search_string="", mirror=N
         if episode:
             source_search += f"E{int(episode):02d}"
 
-    url = f'https://{host}/search'
+    url = f"https://{host}/search"
     headers = {"User-Agent": shared_state.values["user_agent"]}
     data = {"search": source_search}
 
     try:
         r = requests.post(url, headers=headers, data=data, timeout=timeout)
         r.raise_for_status()
-        soup = BeautifulSoup(r.content, 'html.parser')
-        results = soup.find_all('div', class_='article-right')
+        soup = BeautifulSoup(r.content, "html.parser")
+        results = soup.find_all("div", class_="article-right")
     except Exception as e:
         info(f"{hostname}: {search_type} load error: {e}")
-        mark_hostname_issue(hostname, search_type, str(e) if "e" in dir() else "Error occurred")
+        mark_hostname_issue(
+            hostname, search_type, str(e) if "e" in dir() else "Error occurred"
+        )
         return releases
 
     if not results:
@@ -118,13 +135,15 @@ def nk_search(shared_state, start_time, request_from, search_string="", mirror=N
 
     for result in results:
         try:
-            imdb_a = result.select_one('a.imdb')
-            if imdb_a and imdb_a.get('href'):
+            imdb_a = result.select_one("a.imdb")
+            if imdb_a and imdb_a.get("href"):
                 try:
-                    release_imdb_id = re.search(r'tt\d+', imdb_a['href']).group()
+                    release_imdb_id = re.search(r"tt\d+", imdb_a["href"]).group()
                     if imdb_id:
                         if release_imdb_id != imdb_id:
-                            debug(f"{hostname}: IMDb ID mismatch: expected {imdb_id}, found {release_imdb_id}")
+                            debug(
+                                f"{hostname}: IMDb ID mismatch: expected {imdb_id}, found {release_imdb_id}"
+                            )
                             continue
                 except Exception:
                     debug(f"{hostname}: could not extract IMDb ID")
@@ -133,23 +152,25 @@ def nk_search(shared_state, start_time, request_from, search_string="", mirror=N
                 debug(f"{hostname}: could not extract IMDb ID")
                 continue
 
-            a = result.find('a', class_='release-details', href=True)
+            a = result.find("a", class_="release-details", href=True)
             if not a:
                 continue
 
-            sub_title = result.find('span', class_='subtitle')
+            sub_title = result.find("span", class_="subtitle")
             if sub_title:
                 title = sub_title.get_text(strip=True)
             else:
                 continue
 
-            if not shared_state.is_valid_release(title, request_from, search_string, season, episode):
+            if not shared_state.is_valid_release(
+                title, request_from, search_string, season, episode
+            ):
                 continue
 
-            source = urljoin(f'https://{host}', a['href'])
+            source = urljoin(f"https://{host}", a["href"])
 
             mb = 0
-            size_text = get_release_field(result, 'Größe')
+            size_text = get_release_field(result, "Größe")
             if size_text:
                 size_item = extract_size(size_text)
                 mb = shared_state.convert_to_mb(size_item)
@@ -159,45 +180,54 @@ def nk_search(shared_state, start_time, request_from, search_string="", mirror=N
 
             size = mb * 1024 * 1024
 
-            password = ''
-            mirrors_p = result.find('p', class_='mirrors')
+            password = ""
+            mirrors_p = result.find("p", class_="mirrors")
             if mirrors_p:
-                strong = mirrors_p.find('strong')
-                if strong and strong.get_text(strip=True).lower().startswith('passwort'):
+                strong = mirrors_p.find("strong")
+                if strong and strong.get_text(strip=True).lower().startswith(
+                    "passwort"
+                ):
                     nxt = strong.next_sibling
                     if nxt:
                         val = str(nxt).strip()
                         if val:
                             password = val.split()[0]
 
-            date_text = ''
-            p_meta = result.find('p', class_='meta')
+            date_text = ""
+            p_meta = result.find("p", class_="meta")
             if p_meta:
-                spans = p_meta.find_all('span')
+                spans = p_meta.find_all("span")
                 if len(spans) >= 2:
                     date_part = spans[0].get_text(strip=True)
-                    time_part = spans[1].get_text(strip=True).replace('Uhr', '').strip()
+                    time_part = spans[1].get_text(strip=True).replace("Uhr", "").strip()
                     date_text = f"{date_part} / {time_part}"
 
             published = convert_to_rss_date(date_text) if date_text else ""
 
             payload = urlsafe_b64encode(
-                f"{title}|{source}|{mirror}|{mb}|{password}|{release_imdb_id}|{hostname}".encode("utf-8")).decode()
-            link = f"{shared_state.values['internal_address']}/download/?payload={payload}"
-
-            releases.append({
-                "details": {
-                    "title": title,
-                    "hostname": hostname,
-                    "imdb_id": release_imdb_id,
-                    "link": link,
-                    "mirror": mirror,
-                    "size": size,
-                    "date": published,
-                    "source": source
-                },
-                "type": "protected"
-            })
+                f"{title}|{source}|{mirror}|{mb}|{password}|{release_imdb_id}|{hostname}".encode(
+                    "utf-8"
+                )
+            ).decode()
+            link = (
+                f"{shared_state.values['internal_address']}/download/?payload={payload}"
+            )
+
+            releases.append(
+                {
+                    "details": {
+                        "title": title,
+                        "hostname": hostname,
+                        "imdb_id": release_imdb_id,
+                        "link": link,
+                        "mirror": mirror,
+                        "size": size,
+                        "date": published,
+                        "source": source,
+                    },
+                    "type": "protected",
+                }
+            )
         except Exception as e:
             info(e)
             debug(f"{hostname}: error parsing search result: {e}")