add jellyfin api client

jellyfin/api_client.py

@@ -0,0 +1,95 @@
+from settings import settings
+from jellyfin_apiclient_python import JellyfinClient
+from getmac import get_mac_address
+from jellyfin.models import JellyfinMediaItem, JellyfinMediaType
+import logging
+import os
+
+class JellyfinApiClient:
+    def __init__(self):
+        machine_name = os.uname().nodename
+        unique_id = get_mac_address(hostname="localhost")
+
+        self.logger = logging.getLogger('JellyfinApiClient')
+
+        self.logger.info("Connecting to Jellyfin server...")
+        self.client = JellyfinClient()
+        self.client.config.app('jellydisc', '0.0.1', machine_name, unique_id)
+        self.client.config.data['auth.ssl'] = True
+
+        self.client.auth.connect_to_address(settings.jellyfin_server_url)
+        self.client.auth.login(
+            settings.jellyfin_server_url,
+            settings.jellyfin_username,
+            settings.jellyfin_password
+        )
+
+        self.logger.info("Connected to Jellyfin server.")
+
+    def get_current_playback(self) -> JellyfinMediaItem | None:
+        self.logger.info("Fetching current playback information...")
+        sessions = self.client.jellyfin.get_sessions()
+
+        if not sessions:
+            self.logger.info("No active playback found.")
+            return None
+
+        for session in sessions:
+            if session.get('NowPlayingItem'):
+                self.logger.info("Current playback information fetched.")
+                return self.to_model(session['NowPlayingItem'])
+
+        self.logger.info("No active playback found.")
+        return None
+
+    def get_image_url(self, media_id: str) -> str:
+        server_address = settings.jellyfin_server_url.rstrip('/')
+        return f"{server_address}/Items/{media_id}/Images/Primary?maxWidth=300&maxHeight=300"
+
+    def to_model(self, item: dict) -> JellyfinMediaItem:
+        media_type = item.get('Type')
+
+        if media_type == 'Audio':
+            return self.to_music_model(item)
+        elif media_type == 'Episode':
+            return self.to_episode_model(item)
+        elif media_type == 'Movie':
+            return self.to_movie_model(item)
+
+        raise ValueError(f"Unsupported media type: {media_type}")
+
+    def to_music_model(self, item: dict) -> JellyfinMediaItem:
+        media_id = item.get('Id')
+
+        return JellyfinMediaItem(
+            name=item.get('Name'),
+            type=JellyfinMediaType.AUDIO,
+            image_url=self.get_image_url(media_id),
+            metadata={
+                'artist': item.get('AlbumArtist'),
+            }
+        )
+
+    def to_movie_model(self, item: dict) -> JellyfinMediaItem:
+        media_id = item.get('Id')
+
+        return JellyfinMediaItem(
+            name=item.get('Name'),
+            type=JellyfinMediaType.MOVIE,
+            image_url=self.get_image_url(media_id),
+            metadata={}
+        )
+
+    def to_episode_model(self, item: dict) -> JellyfinMediaItem:
+        media_id = item.get('Id')
+
+        return JellyfinMediaItem(
+            name=item.get('Name'),
+            type=JellyfinMediaType.EPISODE,
+            image_url=self.get_image_url(media_id),
+            metadata={
+                'series': item.get('SeriesName'),
+                'season': item.get('ParentIndexNumber'),
+                'episode': item.get('IndexNumber'),
+            }
+        )

jellyfin/models.py

@@ -0,0 +1,13 @@
+from pydantic import BaseModel
+from enum import Enum
+
+class JellyfinMediaType(str, Enum):
+    AUDIO = 'Audio'
+    MOVIE = 'Movie'
+    EPISODE = 'Episode'
+
+class JellyfinMediaItem(BaseModel):
+    name: str
+    type: JellyfinMediaType
+    image_url: str
+    metadata: dict