plexflow 0.0.71__py3-none-any.whl → 0.0.73__py3-none-any.whl

plexflow/core/subtitles/providers/auto_subtitles.py

@@ -8,6 +8,7 @@ from plexflow.utils.retry.utils import execute_until_success
 import time
 from pathlib import Path
 from plexflow.logging.log_setup import logger
+import redis
 
 class AutoSubtitles:
     """
@@ -22,20 +23,20 @@ class AutoSubtitles:
     def __init__(self, imdb_id: str, languages: List[str] = (), **kwargs: Any) -> None:
         self.imdb_id = imdb_id
         self.languages = languages
-        self.redis_hook = kwargs.pop("redis_hook", UniversalRedisHook(redis_conn_id='redis', config_folder='config'))
+        self.redis = kwargs.pop("redis", redis.Redis())
         self.download = kwargs.pop("download", False)
         self.download_folder = Path(kwargs.pop("download_folder", Path(".")))
         self.kwargs = kwargs
     
     def __iter__(self) -> Iterator[OSSSubtitle]:
-        subtitles = get_subtitles(self.imdb_id, self.languages, self.redis_hook, ignore_blacklist=True, **self.kwargs)
+        subtitles = get_subtitles(self.imdb_id, self.languages, self.redis, ignore_blacklist=True, **self.kwargs)
         if len(subtitles) == 0:
             logger.warning(f"No subtitles found for IMDb ID: {self.imdb_id}")
             return
         
         for subtitle in tqdm(subtitles, total=len(subtitles)):
             if self.download:
-                subtitle_path, skipped = execute_until_success(download_subtitle, delay_type='constant', delay=3, max_retries=10, subtitle=subtitle, redis_hook=self.redis_hook, retry_exceptions=[OpenSubtitlesDownloadQuotaReachedException], save_dir=self.download_folder)
+                subtitle_path, skipped = execute_until_success(download_subtitle, delay_type='constant', delay=3, max_retries=10, subtitle=subtitle, r=self.redis, retry_exceptions=[OpenSubtitlesDownloadQuotaReachedException], save_dir=self.download_folder)
                 if not skipped:
                     logger.debug(f"Subtitle downloaded: {subtitle_path}")
                     # time.sleep(1.2)

plexflow/core/subtitles/providers/oss/download.py

@@ -6,8 +6,10 @@ from plexflow.utils.retry.utils import execute_until_success
 from plexflow.utils.hooks.redis import UniversalRedisHook
 from pathlib import Path
 from plexflow.logging.log_setup import logger
+import redis
+import os
 
-def download_subtitle(subtitle: OSSSubtitle, redis_hook: UniversalRedisHook = None, save_dir: Path = Path('.'), skip_exists: bool = True) -> None:
+def download_subtitle(subtitle: OSSSubtitle, r: redis.Redis = None, save_dir: Path = Path('.'), skip_exists: bool = True) -> None:
     """
     Downloads and saves the subtitle file using the OpenSubtitlesManager.
 
@@ -26,8 +28,8 @@ def download_subtitle(subtitle: OSSSubtitle, redis_hook: UniversalRedisHook = No
         return None, True
     else:
         with OpenSubtitlesManager.from_yaml(
-            yaml_file='config/credentials.yaml',
-            redis_hook=redis_hook, 
+            yaml_file=os.getenv("OSS_CREDENTIALS_PATH"),
+            r=r,
         ) as manager:
             filepath.parent.mkdir(parents=True, exist_ok=True)
             metapath.write_text(subtitle.subtitle.to_json())
@@ -43,6 +45,6 @@ def download_subtitles(subtitles: List[OSSSubtitle], **kwargs) -> None:
     Returns:
         None
     """
-    redis_hook = kwargs.pop("redis_hook", UniversalRedisHook(redis_conn_id='redis', config_folder='config'))
+    r = kwargs.pop("redis", redis.Redis())
     for subtitle in subtitles:
-        execute_until_success(download_subtitle, delay_type='constant', delay=3, max_retries=10, subtitle=subtitle, redis_hook=redis_hook, retry_exceptions=[OpenSubtitlesDownloadQuotaReachedException])
+        execute_until_success(download_subtitle, delay_type='constant', delay=3, max_retries=10, subtitle=subtitle, r=r, retry_exceptions=[OpenSubtitlesDownloadQuotaReachedException])

plexflow/core/subtitles/providers/oss/oss.py

@@ -22,11 +22,12 @@ from plexflow.core.subtitles.providers.oss.utils.download_client import Download
 from plexflow.core.subtitles.providers.oss.utils.languages import language_codes
 from plexflow.utils.hooks.redis import UniversalRedisHook
 from plexflow.logging.log_setup import logger
+import redis
 
 class OpenSubtitles:
     """OpenSubtitles REST API Wrapper."""
 
-    def __init__(self, user_agent: str, api_key: str, redis_hook: UniversalRedisHook = None):
+    def __init__(self, user_agent: str, api_key: str, r: redis.Redis = None):
         """Initialize the OpenSubtitles object.
 
         :param api_key:
@@ -41,13 +42,13 @@ class OpenSubtitles:
         self.downloads_dir = "."
         self.user_downloads_remaining = 0
         self.reset_time = None
-        self.redis_hook = redis_hook
+        self.redis = r
         
     @property
     def cached_token(self):
         try:
-            if self.redis_hook:
-                return self.redis_hook.get(f"@opensubtitles/{self.api_key}/token").decode("utf-8")
+            if self.redis:
+                return self.redis.get(f"@opensubtitles/{self.api_key}/token").decode("utf-8")
             else:
                 return None
         except Exception as e:
@@ -59,7 +60,7 @@ class OpenSubtitles:
     def cached_token(self, token: str):
         try:
             logger.debug(f"caching token: {token}")
-            self.redis_hook.set(f"@opensubtitles/{self.api_key}/token", token, ex=60*60)
+            self.redis.set(f"@opensubtitles/{self.api_key}/token", token, ex=60*60)
         except Exception as e:
             logger.error(f"Error setting cached token: {e}")
 

plexflow/core/subtitles/providers/oss/search.py

@@ -5,9 +5,10 @@ from plexflow.core.subtitles.providers.oss.oss_subtitle import OSSSubtitle
 from typing import Any, List
 from contextlib import contextmanager, ExitStack
 from plexflow.utils.hooks.redis import UniversalRedisHook
+import redis
 
 @contextmanager
-def open_subtitles_manager(credentials_path: str, redis_hook: UniversalRedisHook = None, **kwargs: Any):
+def open_subtitles_manager(credentials_path: str, r: redis.Redis = None, **kwargs: Any):
     """
     Context manager for managing the OpenSubtitlesManager instance.
 
@@ -22,12 +23,12 @@ def open_subtitles_manager(credentials_path: str, redis_hook: UniversalRedisHook
     with ExitStack() as stack:
         manager = stack.enter_context(OpenSubtitlesManager.from_yaml(
             yaml_file=credentials_path,
-            redis_hook=redis_hook,
+            r=r,
             **kwargs
         ))
         yield manager
 
-def get_subtitles(imdb_id: str, languages: List[str] = (), redis_hook: UniversalRedisHook = None, ignore_blacklist: bool = False, **kwargs) -> List[OSSSubtitle]:
+def get_subtitles(imdb_id: str, languages: List[str] = (), r: redis.Redis = None, ignore_blacklist: bool = False, **kwargs) -> List[OSSSubtitle]:
     """
     Retrieves subtitles using OpenSubtitlesManager.
 
@@ -40,7 +41,7 @@ def get_subtitles(imdb_id: str, languages: List[str] = (), redis_hook: Universal
     """
     with open_subtitles_manager(
         credentials_path=kwargs.pop("credentials_path"),
-        redis_hook=redis_hook,
+        r=r,
         ignore_blacklist=ignore_blacklist,
     ) as manager:
         subtitles = manager.search(

plexflow/core/subtitles/providers/oss/unlimited_oss.py

@@ -53,7 +53,7 @@ class OpenSubtitlesManager:
     REDIS_DEFAULT_PORT = 6379
     DEFAULT_TTL = 60*60  # 1 hour
 
-    def __init__(self, credentials: List[Dict[str, str]], redis_hook: UniversalRedisHook = None, **kwargs):
+    def __init__(self, credentials: List[Dict[str, str]], r: redis.Redis = None, **kwargs):
         """
         Initializes an instance of the OpenSubtitlesManager class.
 
@@ -64,7 +64,7 @@ class OpenSubtitlesManager:
             redis_port (int, optional): The port of the Redis server used for blacklisting credentials. Defaults to 6379.
         """
         self.credentials = credentials
-        self.redis_hook = redis_hook
+        self.redis = r
         self.current_instance = None
         self.current_credential = None
         self.ignore_blacklist = kwargs.get('ignore_blacklist', False)
@@ -72,7 +72,7 @@ class OpenSubtitlesManager:
         self.instances = {}
         
     @classmethod
-    def from_yaml(cls, yaml_file: str, redis_hook: UniversalRedisHook = None, **kwargs) -> 'OpenSubtitlesManager':
+    def from_yaml(cls, yaml_file: str, r: redis.Redis = None, **kwargs) -> 'OpenSubtitlesManager':
         """
         Creates an instance of the OpenSubtitlesManager class from a YAML file.
 
@@ -86,7 +86,7 @@ class OpenSubtitlesManager:
         """
         with open(yaml_file, 'r') as file:
             credentials = yaml.safe_load(file)
-        return cls(credentials=credentials, redis_hook=redis_hook, **kwargs)
+        return cls(credentials=credentials, r=r, **kwargs)
 
     def __enter__(self) -> OpenSubtitles:
         """
@@ -181,7 +181,7 @@ class OpenSubtitlesManager:
         if api_key in self.instances:
             return self.instances[api_key]
         
-        instance = OpenSubtitles(user_agent, api_key, redis_hook=self.redis_hook)
+        instance = OpenSubtitles(user_agent, api_key, r=self.redis)
         instance.login(username, password)
         
         # Cache the instance for later reuse
@@ -201,7 +201,7 @@ class OpenSubtitlesManager:
         """
         key = self.generate_credential_key(credential)
         self.logger.debug(f"Generated credential key: {key}")
-        return self.redis_hook.get_conn().exists(key)
+        return self.redis.exists(key)
 
     def blacklist_credential(self):
         """
@@ -209,7 +209,7 @@ class OpenSubtitlesManager:
 
         """
         key = self.generate_credential_key(self.current_credential)
-        self.redis_hook.get_conn().setex(key, self.DEFAULT_TTL, 'blacklisted')
+        self.redis.setex(key, self.DEFAULT_TTL, 'blacklisted')
         self.logger.debug(f"Credential blacklisted for {self.DEFAULT_TTL} seconds: {self.current_credential}")
         self.logger.debug(f"Blacklisted credential key: {key}")
 

{plexflow-0.0.71.dist-info → plexflow-0.0.73.dist-info}/METADATA

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

{plexflow-0.0.71.dist-info → plexflow-0.0.73.dist-info}/RECORD

@@ -301,29 +301,29 @@ plexflow/core/subtitles/providers/__pycache__/__init__.cpython-311.pyc,sha256=2z
 plexflow/core/subtitles/providers/__pycache__/__init__.cpython-312.pyc,sha256=gp9rHSu09WohhuFlVII26NTnqB9maGtEsjc8rKQ7pvo,171
 plexflow/core/subtitles/providers/__pycache__/auto_subtiltes.cpython-311.pyc,sha256=Coddhp2j6Ay3oA1H5qQEHbn1qqpBLrHaINC8K-TaZqI,3857
 plexflow/core/subtitles/providers/__pycache__/auto_subtitles.cpython-311.pyc,sha256=BEiGR-WcFPy0Lp4bgA8ykjs081bpT5m0L3sqy6Kqfdg,3893
-plexflow/core/subtitles/providers/__pycache__/auto_subtitles.cpython-312.pyc,sha256=ygF64q4km2n_9KSb7tZwBbDUiMQqKvd7HH8wGI74Vuc,3496
-plexflow/core/subtitles/providers/auto_subtitles.py,sha256=CWHpQ_-YD_c0v9X2NCbPAUVgpOf6_G7SOkilsqlxXpg,2268
+plexflow/core/subtitles/providers/__pycache__/auto_subtitles.cpython-312.pyc,sha256=JqzDzkfmfCe_2Oa_XKWtvGAb7s2QbYP3ExS0xQGXB6c,3463
+plexflow/core/subtitles/providers/auto_subtitles.py,sha256=lq7pEBBkrrkVjieB7uPs89v-SDlrOGbMhUzkA0pzii0,2200
 plexflow/core/subtitles/providers/oss/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 plexflow/core/subtitles/providers/oss/__pycache__/__init__.cpython-311.pyc,sha256=fisyzWOS6ZR67F7A-k7WEaapRbm2h0mapOixVKgavqs,187
 plexflow/core/subtitles/providers/oss/__pycache__/__init__.cpython-312.pyc,sha256=H2fhQnTvVsfRSfrDSReYATGTisbG-QEA6P-AHi8y-1U,175
 plexflow/core/subtitles/providers/oss/__pycache__/datatypes.cpython-311.pyc,sha256=wCE0qHCaISXvsiDdrakvfbW8_Fww3JeG8ep2wmfwgQ4,8771
 plexflow/core/subtitles/providers/oss/__pycache__/download.cpython-311.pyc,sha256=qyfLK7z1Yb0r5aM8KvPLeu0P1hflX5zqHEvfLtmF_hw,3733
-plexflow/core/subtitles/providers/oss/__pycache__/download.cpython-312.pyc,sha256=1PiPdkgccezU1H4_JsSgl7DFlpt6ZECTa-usrx2oF14,3330
+plexflow/core/subtitles/providers/oss/__pycache__/download.cpython-312.pyc,sha256=AJ3vVJBQKuvE0xI9CrhwXDyKm2bG-gfZWNbWfn9PWgg,3307
 plexflow/core/subtitles/providers/oss/__pycache__/oss.cpython-311.pyc,sha256=tbcGIfaslOGnfjwoko8TSKIGRRO4sM86Gbmy7OG9occ,21609
-plexflow/core/subtitles/providers/oss/__pycache__/oss.cpython-312.pyc,sha256=TNur-bB1Smaeu7ap_TCy60H_3a-pLgpJK9uVOdqm2fI,19661
+plexflow/core/subtitles/providers/oss/__pycache__/oss.cpython-312.pyc,sha256=UXRE-8qiUAEkB55aYFtWb7WL8jHXozdIDGHZWkyQKRM,19694
 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=0PePMgbDM2KZPA7JUib_JjKtDziyICdBDU6xBc09Lz8,2287
 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=37GzKOOE0yRgBpIiW06YiipVR3JpM7w8nXdq1DJSoKs,2618
+plexflow/core/subtitles/providers/oss/__pycache__/search.cpython-312.pyc,sha256=nSqaXHXI1Fe9mncp492AhLr444SMze-6QrZkdz94wr8,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=XaHVt2HU5IjWIj4RIilEolHYGKjzikDvfc12tudXMlk,11772
+plexflow/core/subtitles/providers/oss/__pycache__/unlimited_oss.cpython-312.pyc,sha256=Z4akT251BicN1_OsJ8VWq-7b5EoPyHL-JfalKVEi0xA,11705
 plexflow/core/subtitles/providers/oss/datatypes.py,sha256=7YvjS8a3riEJ4yqLKzTRv5g6vftcrcoYKGcBoq0MCLo,3620
-plexflow/core/subtitles/providers/oss/download.py,sha256=s4y3jAQnCIzugwYvpcdJg5kRgzUwopvyWItZl8Adimg,2209
+plexflow/core/subtitles/providers/oss/download.py,sha256=xEnmslFQnHhk8gDw35ePDGF-MZo47UUvivpHFhCzhsM,2121
 plexflow/core/subtitles/providers/oss/old.py,sha256=IoZ7iIRs3FoxxJgzGrgMGNJpVXqZ3aAaadjKJJ3Lw0o,5566
-plexflow/core/subtitles/providers/oss/oss.py,sha256=FXuuxc01KManZEF0PTV5vNMZVT4nhpm7T83H0xcZ6Q0,15953
+plexflow/core/subtitles/providers/oss/oss.py,sha256=m0UGVMs50-et8_RfYCOOJXPDRabBZOW8UyLiPeq83AQ,15921
 plexflow/core/subtitles/providers/oss/oss_subtitle.py,sha256=SuFv90Z_KUELHnAm85SpZNsH0tXKJMMgFKODYgjDmzc,879
-plexflow/core/subtitles/providers/oss/search.py,sha256=MaDmrKIC-lZO_THXPaDVgYNQ-ZVkmQdfZzfuujUeV5w,1920
-plexflow/core/subtitles/providers/oss/unlimited_oss.py,sha256=Ra-9uqCiAgEo_wQ-GZRkj_ocIw5mhU_0iR89TY653OU,9672
+plexflow/core/subtitles/providers/oss/search.py,sha256=VKMVgaZ67Juxlvf7IRqpo1QYKVKAaJg8Zhj-NQhjoy0,1865
+plexflow/core/subtitles/providers/oss/unlimited_oss.py,sha256=7PN6N5-1Pc9TNQqmnIPX5Q9ZQhEAnRTrO2ExlE2xOWc,9562
 plexflow/core/subtitles/providers/oss/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 plexflow/core/subtitles/providers/oss/utils/__pycache__/__init__.cpython-311.pyc,sha256=nBHOv53UkuFjaPU3Zih_W5tvdRYEmZOeySSrsa3qsGo,193
 plexflow/core/subtitles/providers/oss/utils/__pycache__/__init__.cpython-312.pyc,sha256=nWQpxGf6NgdB_p8RS6e0DO4jYZ9h8xjAc8GGiJUs8wk,181
@@ -643,7 +643,7 @@ plexflow/utils/video/__pycache__/audio.cpython-312.pyc,sha256=vXBnJwWgTDFdixMBs-
 plexflow/utils/video/__pycache__/subtitle.cpython-312.pyc,sha256=PCjpCLydGXaRsQy6cikhgsEs8WlComfOoYPiLFqfVMA,2515
 plexflow/utils/video/audio.py,sha256=tJ_lNwcjVuBQYD5cYOlXpr__eh8-hnReIgNRgIYOpqo,3380
 plexflow/utils/video/subtitle.py,sha256=LOGONGxs_RzmqtGP-DBKreOzS1eUFEKo75Q6AfnavW0,1290
-plexflow-0.0.71.dist-info/METADATA,sha256=UPDJWMRiHp1i8d-xL_og6-707uniskaOv3bvZCYhbU4,2954
-plexflow-0.0.71.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
-plexflow-0.0.71.dist-info/entry_points.txt,sha256=A-_w_iUmjrSgDtOcX-G5H3NboAAZNzwiFoQbBf4bVjg,1302
-plexflow-0.0.71.dist-info/RECORD,,
+plexflow-0.0.73.dist-info/METADATA,sha256=eEbSSAHrVPA3tZJADM2F-spWyEOBOrHEI3X4TQ0Wp2U,2954
+plexflow-0.0.73.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
+plexflow-0.0.73.dist-info/entry_points.txt,sha256=A-_w_iUmjrSgDtOcX-G5H3NboAAZNzwiFoQbBf4bVjg,1302
+plexflow-0.0.73.dist-info/RECORD,,