Merge pull request #3 from mov-cli/feat/plex

Feat/plex

Author: ananasmoe

mov_cli_ms/__init__.py

@@ -5,13 +5,15 @@
     from mov_cli.plugins import PluginHookData
 
 from .jellyfin.scraper import *
+from .plex.scraper import *
 
 plugin: PluginHookData = {
     "version": 1, # plugin hook version
     "package_name": "mov-cli-ms", # pypi package name
     "scrapers": {
         "DEFAULT": JellyfinScraper, 
-        "jellyfin": JellyfinScraper
+        "jellyfin": JellyfinScraper,
+        "plex": PlexScraper
     }
 }
 

mov_cli_ms/plex/scraper.py

@@ -8,24 +8,89 @@
     from mov_cli.http_client import HTTPClient
     from mov_cli.scraper import ScraperOptionsT
 
-from mov_cli import Single, Multi, Metadata
+from dataclasses import dataclass, field
+
+from mov_cli import Single, Multi, Metadata, MetadataType
 
 from mov_cli.scraper import Scraper
 from mov_cli.utils import EpisodeSelector
 
-__all__ = ("PlexScraper", )
+from plexapi.myplex import MyPlexAccount
+from plexapi.server import PlexServer
+from plexapi.exceptions import TwoFactorRequired
+from plexapi.video import Movie, Show
+
+__all__ = ("PlexScraper", "PlexMetadata",)
+
+@dataclass
+class PlexMetadata(Metadata):
+    id: int
+    video: Movie | Show = field(default = None)
 
 class PlexScraper(Scraper):
     def __init__(self, config: Config, http_client: HTTPClient, options: Optional[ScraperOptionsT] = None) -> None:
-        self.base_url = ...
+        env_config = config.get_env_config()
+
+        self.base_url = env_config("PLEX_SERVER_ID", default = None, cast = str)
+        self.username = env_config("PLEX_USERNAME", default = None, cast = str)
+        self.password = env_config("PLEX_PASSWORD", default = None, cast = str)
+
+        self.plex = self.__auth()
 
         super().__init__(config, http_client, options)
 
-    def search(self, query: str, limit: int = 20) -> Generator[Metadata, Any, None]:
-        ...
 
-    def scrape_metadata_episodes(self, metadata: Metadata) -> Dict[int, int] | Dict[None, int]:
-        ...
-    
-    def scrape(self, metadata: Metadata, episode: EpisodeSelector) -> Single | Multi:
-        ...
+    def search(self, query: str, limit: int = 20) -> Generator[PlexMetadata, Any, None]:
+        videos = self.plex.search(query, limit = limit)
+
+        for _, video in enumerate(videos):
+            if video.TYPE in ["movie", "show"]:
+                yield PlexMetadata(
+                    id = _,
+                    title = video.title,
+                    type = MetadataType.SINGLE if "movie" == video.TYPE else MetadataType.MULTI,
+                    year = video.year,
+                    video = video
+                )
+
+    def scrape_episodes(self, metadata: PlexMetadata) -> Dict[int, int] | Dict[None, int]:
+        scraped_episodes = {}
+
+        seasons = metadata.video.seasons()
+
+        for season in seasons:
+            scraped_episodes[season.seasonNumber] = len(season.episodes())
+
+        return scraped_episodes
+
+    def scrape(self, metadata: PlexMetadata, episode: EpisodeSelector) -> Single | Multi:
+        if metadata.type == MetadataType.MULTI:
+            epi = metadata.video.episode(season = episode.season, episode = episode.episode)
+
+            return Multi(
+                url = self.__make_url(epi),
+                title = metadata.title,
+                episode = episode
+            )
+        
+        return Single(
+            url = self.__make_url(metadata.video),
+            title = metadata.title,
+            year = metadata.year
+        )
+
+    def __make_url(self, item) -> str:
+        key = next(item.iterParts()).key
+
+        return item._server.url(f"{key}?download=1&X-Plex-Token={self.plex.account().authToken}")
+
+    def __auth(self) -> PlexServer:
+        try:
+            account = MyPlexAccount(self.username, self.password)
+        except TwoFactorRequired:
+            code = self.options.get("2FA")
+            account = MyPlexAccount(self.username, self.password, code=code)
+
+        plex = account.resource(self.base_url).connect()
+
+        return plex

pyproject.toml

@@ -26,7 +26,7 @@ dependencies = [
     "requests",
     "importlib-metadata; python_version<'3.8'",
 
-    "mov-cli>=4.3"
+    "plexapi"
 ]
 
 dynamic = ["version"]