plexflow 0.0.129__py3-none-any.whl → 0.0.131__py3-none-any.whl

plexflow/core/context/partials/show.py

@@ -158,3 +158,16 @@ class Show(PartialContext):
     @property
     def selected_episode(self) -> PlexEpisodeMetadata:
         return self.get('episodes/selected')
+    
+    def update_episode_imdb_codes(self, season: int, codes: dict):
+        self.set(f"episodes/season/{season}/imdb/codes", codes)
+    
+    def episode_imdb_codes(self, season: int) -> dict:
+        return self.get(f"episodes/season/{season}/imdb/codes")
+
+    @property
+    def selected_episode_imdb_code(self) -> str:
+        episode = self.selected_episode
+        season = episode.parentIndex
+        codes = self.episode_imdb_codes(season=season)
+        return codes.get(episode.index)

plexflow/utils/imdb/episodes.py

@@ -0,0 +1,60 @@
+import requests
+from bs4 import BeautifulSoup
+from typing import Dict, Optional
+from plexflow.utils.imdb.imdb_codes import extract_imdb_code
+
+def get_imdb_episode_codes(imdb_title_id: str, season_number: int) -> Dict[int, Optional[str]]:
+    """
+    Fetches IMDb episode codes for a given title ID and season number.
+
+    Args:
+        imdb_title_id (str): The IMDb title ID (e.g., 'tt3581920').
+        season_number (int): The season number to retrieve episodes for.
+
+    Returns:
+        Dict[int, Optional[str]]: A dictionary where keys are episode numbers
+                                  and values are their corresponding IMDb codes.
+                                  Returns an empty dictionary if no episodes are found or
+                                  an error occurs.
+    """
+    if not imdb_title_id.startswith('tt'):
+        print("Warning: IMDb title ID should start with 'tt'.")
+        return {}
+
+    url = f"https://www.imdb.com/title/{imdb_title_id}/episodes/?season={season_number}&ref_=ttep"
+    headers = {
+        'User-Agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/605.1.15 (KHTML, like Gecko) Version/18.4 Safari/605.1.15',
+    }
+
+    try:
+        r = requests.get(url=url, headers=headers, timeout=10) # Added timeout for robustness
+        r.raise_for_status()  # Raise an exception for HTTP errors (4xx or 5xx)
+    except requests.exceptions.RequestException as e:
+        print(f"Error fetching URL {url}: {e}")
+        return {}
+
+    soup = BeautifulSoup(r.text, 'html.parser')
+
+    # Select all <a> tags within elements that have class 'episode-item-wrapper'
+    # and whose href attribute starts with '/title/'.
+    links = soup.select('.episode-item-wrapper a[href^="/title/"]')
+
+    episodes: Dict[int, Optional[str]] = {}
+    seen_urls = set()
+    current_episode_number = 1
+
+    for link in links:
+        href = link['href']
+        # IMDb often lists the same episode link multiple times (e.g., image and title link).
+        # We only want to process each unique episode URL once.
+        if href not in seen_urls:
+            seen_urls.add(href)
+            # The extract_imdb_code function should return the IMDb code directly,
+            # not an iterator, based on common usage. If it returns an iterator,
+            # `next(..., None)` is correct.
+            episode_imdb_code = next(extract_imdb_code(href), None)
+            if episode_imdb_code: # Only add if a valid IMDb code was extracted
+                episodes[current_episode_number] = episode_imdb_code
+                current_episode_number += 1
+
+    return episodes

{plexflow-0.0.129.dist-info → plexflow-0.0.131.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: plexflow
-Version: 0.0.129
+Version: 0.0.131
 Summary: A short description of the package.
 License: MIT
 Keywords: keyword1,keyword2,keyword3

{plexflow-0.0.129.dist-info → plexflow-0.0.131.dist-info}/RECORD

@@ -32,7 +32,7 @@ plexflow/core/context/partials/ids.py,sha256=QoQ6FbX1OIWrE-iuz-G6kSzBlTt1_I1jyfl
 plexflow/core/context/partials/movie.py,sha256=VXQ2SspFgGSRgDefg4VlHrH2fns3KRuKlU72ps6527o,3861
 plexflow/core/context/partials/movie_assets.py,sha256=qjZTs-lpPfZkQQSKm6CB4aeECX5_YzOom51PxZzmnts,1913
 plexflow/core/context/partials/reports.py,sha256=0W58RwK3VSsVHbF0rhvMNNlZZr01eutwermyvdeEZIs,810
-plexflow/core/context/partials/show.py,sha256=lfjh1mVg49SEFzMSLfseDs4MgLKLx_8_ltHtlWoHbUc,5085
+plexflow/core/context/partials/show.py,sha256=SpNGCRV8TYyqWc3_gZekFndvENJSy9sabDtilW_grKg,5580
 plexflow/core/context/partials/subtitles.py,sha256=0NhKGkP-sArQswuSyA7puRSjjoobF-3Ah7Pd39QkgTU,535
 plexflow/core/context/partials/tgx_batch.py,sha256=TduB09oBOQ8CtmPYsHIeNe7AI-ypKw21zQAX-7qktEs,859
 plexflow/core/context/partials/tgx_context.py,sha256=_FuhOvKsFqi_uynHxgC9_QIR2CfYmz-uJCRFtGFJmXI,1641
@@ -350,7 +350,7 @@ plexflow/core/subtitles/providers/oss/__pycache__/oss.cpython-312.pyc,sha256=XhZ
 plexflow/core/subtitles/providers/oss/__pycache__/oss_subtitle.cpython-311.pyc,sha256=IfUEII4Khd7b4vs1Ag73irbmoulk1f9uKOX9xI0vtlo,2396
 plexflow/core/subtitles/providers/oss/__pycache__/oss_subtitle.cpython-312.pyc,sha256=YJjcasSbQna30MtyKApyFrU4VFwNOxh0awY0565RPkE,2576
 plexflow/core/subtitles/providers/oss/__pycache__/search.cpython-311.pyc,sha256=Hb6ZX9B0wjeOvOMNaw-eZnPBITsINQ3Cs8cYebq-dKs,3038
-plexflow/core/subtitles/providers/oss/__pycache__/search.cpython-312.pyc,sha256=QCXsAPqp5G4oo2S-GvRaDmOQF1eVnvlyj4YBgLRSJCI,2662
+plexflow/core/subtitles/providers/oss/__pycache__/search.cpython-312.pyc,sha256=nzKPBAPeNoX7ApIMz3Z-aGAfmnEEg2Bz_7KRUm-nAIw,2662
 plexflow/core/subtitles/providers/oss/__pycache__/unlimited_oss.cpython-311.pyc,sha256=OUvBlMRcl1mDpT9UnhZl9g9YZgfJj5Y8Gff2oIto1V8,12587
 plexflow/core/subtitles/providers/oss/__pycache__/unlimited_oss.cpython-312.pyc,sha256=bU2sKm2qYOI-8AO25Xswuwv0AAPRtX04DTdYq3r-K-I,11756
 plexflow/core/subtitles/providers/oss/datatypes.py,sha256=7YvjS8a3riEJ4yqLKzTRv5g6vftcrcoYKGcBoq0MCLo,3620
@@ -641,6 +641,7 @@ plexflow/utils/imdb/__pycache__/__init__.cpython-311.pyc,sha256=Om33PdqDky-6oW50
 plexflow/utils/imdb/__pycache__/__init__.cpython-312.pyc,sha256=HB4wxnIqRG2mIgj9A1l6jLVTmlCs7LF2nsnlDEdIh8Q,155
 plexflow/utils/imdb/__pycache__/imdb_codes.cpython-311.pyc,sha256=pGACYM3C88N41gMgFGz8L0evmhavIoineUeK8XPv3oU,4396
 plexflow/utils/imdb/__pycache__/imdb_codes.cpython-312.pyc,sha256=PqmomwWgbBJ4HdQHSwg2FIesLrR72Tbs58uoZWrjqQc,4092
+plexflow/utils/imdb/episodes.py,sha256=q5Mcsdr2i-zRX8Ce23gLqTi0G8-XRX9ZHpTFb5Iu2Ps,2602
 plexflow/utils/imdb/imdb_codes.py,sha256=aIECE9IEFSYbY3A-erONQiFUqy36JrH2hhcM3qWObk0,3028
 plexflow/utils/llm/mistral.py,sha256=OX_ogd7i3xzh7959O65oGzzfhuGRTFWsUwwmkShH7fs,4521
 plexflow/utils/pubsub/__pycache__/consume.cpython-312.pyc,sha256=QlvfpCEFCf53lhve-f1QdeNLhiVXRVkqtowp4n8is9w,3780
@@ -713,7 +714,7 @@ plexflow/utils/video/__pycache__/audio.cpython-312.pyc,sha256=kmzGDCHSC1hWyHwRut
 plexflow/utils/video/__pycache__/subtitle.cpython-312.pyc,sha256=PCjpCLydGXaRsQy6cikhgsEs8WlComfOoYPiLFqfVMA,2515
 plexflow/utils/video/audio.py,sha256=Pd8OuQHX2QN-lc5iYkB0Vo1OEHmTcvDYH-uKud1f1q4,5262
 plexflow/utils/video/subtitle.py,sha256=qPvvBjlPj0fynJJvGJgGeKt9ey26R-cF6EoLaYt9iXU,1333
-plexflow-0.0.129.dist-info/METADATA,sha256=OtWB_9gAVCtNivzs5zq1g8IAYt1374qQyCHOpOoKsEY,2971
-plexflow-0.0.129.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
-plexflow-0.0.129.dist-info/entry_points.txt,sha256=9RJC3ikOQORHNOn573EdwJOBUnFU_4EGHbtNUM5pjjY,1557
-plexflow-0.0.129.dist-info/RECORD,,
+plexflow-0.0.131.dist-info/METADATA,sha256=TtF9s3LrMhFwP08ejAELCphKk0mA8LTeXzBOM1ROBsE,2971
+plexflow-0.0.131.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
+plexflow-0.0.131.dist-info/entry_points.txt,sha256=9RJC3ikOQORHNOn573EdwJOBUnFU_4EGHbtNUM5pjjY,1557
+plexflow-0.0.131.dist-info/RECORD,,