59 lines
1.8 KiB
Python
59 lines
1.8 KiB
Python
"""set metadata to episodes"""
|
|
|
|
from datetime import datetime
|
|
|
|
from src.connect import Jellyfin, TubeArchivist
|
|
|
|
|
|
class Episode:
|
|
"""interact with an single episode"""
|
|
|
|
def __init__(self, youtube_id, jf_id):
|
|
self.youtube_id = youtube_id
|
|
self.jf_id = jf_id
|
|
|
|
def sync(self):
|
|
"""sync episode metadata"""
|
|
ta_video = self.get_ta_video()
|
|
self.update_metadata(ta_video)
|
|
self.update_artwork(ta_video)
|
|
|
|
def get_ta_video(self):
|
|
"""get video metadata from ta"""
|
|
path = f"/video/{self.youtube_id}"
|
|
ta_video = TubeArchivist().get(path)
|
|
|
|
return ta_video
|
|
|
|
def update_metadata(self, ta_video):
|
|
"""update jellyfin metadata from item_id"""
|
|
published = ta_video.get("published")
|
|
published_date = datetime.strptime(published, "%d %b, %Y")
|
|
data = {
|
|
"Id": self.jf_id,
|
|
"Name": ta_video.get("title"),
|
|
"Genres": [],
|
|
"Tags": [],
|
|
"ProductionYear": published_date.year,
|
|
"ProviderIds": {},
|
|
"ParentIndexNumber": published_date.year,
|
|
"PremiereDate": published_date.isoformat(),
|
|
"Overview": self._get_desc(ta_video),
|
|
}
|
|
path = f"Items/{self.jf_id}"
|
|
Jellyfin().post(path, data)
|
|
|
|
def update_artwork(self, ta_video):
|
|
"""update episode artwork in jf"""
|
|
thumb_base64 = TubeArchivist().get_thumb(ta_video.get("vid_thumb_url"))
|
|
path = f"Items/{self.jf_id}/Images/Primary"
|
|
Jellyfin().post_img(path, thumb_base64)
|
|
|
|
def _get_desc(self, ta_video):
|
|
"""get description"""
|
|
raw_desc = ta_video.get("description").replace("\n", "<br>")
|
|
if len(raw_desc) > 500:
|
|
return raw_desc[:500] + " ..."
|
|
|
|
return raw_desc
|