130 lines
4.2 KiB
Python
130 lines
4.2 KiB
Python
import logging
|
|
import os
|
|
from typing import Any
|
|
from urllib.parse import unquote
|
|
|
|
import feedparser
|
|
import requests
|
|
from dateutil.parser import ParserError, parse
|
|
from podcasts.models import PodcastEpisode
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
# TODO This should be configurable in settings or per deploy
|
|
PODCAST_DATE_FORMAT = "YYYY-MM-DD"
|
|
|
|
def parse_duration(d):
|
|
if not d:
|
|
return None
|
|
if d.isdigit():
|
|
return int(d)
|
|
parts = [int(p) for p in d.split(":")]
|
|
while len(parts) < 3:
|
|
parts.insert(0, 0)
|
|
h, m, s = parts
|
|
return h * 3600 + m * 60 + s
|
|
|
|
def fetch_metadata_from_rss(uri: str) -> dict[str, Any]:
|
|
log_context = {"mopidy_uri": uri, "media_type": "Podcast"}
|
|
podcast_data: dict[str, Any] = {}
|
|
|
|
rss_url = uri.split("#")[0].split("podcast+")[1]
|
|
target_guid = uri.split("#")[1]
|
|
|
|
log_context["rss_url"] = rss_url
|
|
log_context["target_guid"] = target_guid
|
|
|
|
try:
|
|
resp = requests.get(rss_url, timeout=10)
|
|
feed = feedparser.parse(resp.text)
|
|
except IndexError:
|
|
logger.warning("Tried to parse uri as RSS feed, but no target found", extra=log_context)
|
|
return podcast_data
|
|
|
|
podcast_publisher = getattr(feed.feed, "itunes_publisher", "")
|
|
try:
|
|
podcast_owner = feed.feed.itunes_owner.get("name") if isinstance(feed.feed.itunes_owner, dict) else feed.feed.itunes_owner
|
|
podcast_other = feed.feed.get("managingeditor") or feed.feed.get("copyright")
|
|
except AttributeError:
|
|
podcast_owner = None
|
|
podcast_other = None
|
|
|
|
podcast_data = {
|
|
"podcast_name": getattr(feed.feed, "title", ""),
|
|
# "podcast_description": getattr(feed.feed, "description", ""),
|
|
# "podcast_link": getattr(feed.feed, "link", ""),
|
|
"podcast_producer": podcast_publisher or podcast_owner or podcast_other
|
|
}
|
|
|
|
for entry in feed.entries:
|
|
if entry.get("guid") == target_guid:
|
|
logger.info("🎧 Episode found in RSS feed", extra=log_context)
|
|
podcast_data["title"] = entry.title
|
|
podcast_data["episode_num"] = int(entry.get("itunes_episode", 0))
|
|
podcast_data["pub_date"] = parse(entry.get("published", None))
|
|
podcast_data["base_run_time_seconds"] = parse_duration(entry.get("itunes_duration", None))
|
|
# podcast_data["description"] = entry.get("description", None)
|
|
# podcast_data["episode_url"] = entry.enclosures[0].href if entry.get("enclosures") else None
|
|
return podcast_data
|
|
else:
|
|
logger.info("Episode not found in RSS feed.")
|
|
|
|
|
|
def parse_mopidy_uri(uri: str) -> dict[str, Any]:
|
|
podcast_data: dict[str, Any] = {}
|
|
|
|
logger.debug(f"Parsing URI: {uri}")
|
|
if "podcast+https" in uri:
|
|
return fetch_metadata_from_rss(uri)
|
|
|
|
|
|
parsed_uri = os.path.splitext(unquote(uri))[0].split("/")
|
|
|
|
podcast_data = {
|
|
"title": parsed_uri[-1],
|
|
"episode_num": None,
|
|
"podcast_name": parsed_uri[-2].strip(),
|
|
"pub_date": None,
|
|
}
|
|
|
|
|
|
episode_str = parsed_uri[-1]
|
|
episode_num = None
|
|
episode_num_pad = 0
|
|
|
|
try:
|
|
# Without episode numbers the date will lead
|
|
podcast_data["pub_date"] = parse(episode_str[0:10])
|
|
except ParserError:
|
|
podcast_data["episode_num"] = int(episode_str.split("-")[0])
|
|
episode_num_pad = len(str(podcast_data["episode_num"])) + 1
|
|
|
|
try:
|
|
# Beacuse we have epsiode numbers on
|
|
podcast_data["pub_date"] = parse(
|
|
episode_str[
|
|
episode_num_pad : len(PODCAST_DATE_FORMAT)
|
|
+ episode_num_pad
|
|
]
|
|
)
|
|
except ParserError:
|
|
pub_date = ""
|
|
|
|
gap_to_strip = 0
|
|
if podcast_data["pub_date"]:
|
|
gap_to_strip += len(PODCAST_DATE_FORMAT)
|
|
if podcast_data["episode_num"]:
|
|
gap_to_strip += episode_num_pad
|
|
|
|
podcast_data["title"] = episode_str[gap_to_strip:].replace("-", " ").strip()
|
|
|
|
return podcast_data
|
|
|
|
|
|
def get_or_create_podcast(post_data: dict) -> PodcastEpisode:
|
|
logger.info("Looking up podcast", extra={"post_data": post_data, "media_type": "Podcast"})
|
|
mopidy_uri = post_data.get("mopidy_uri", "")
|
|
parsed_data = parse_mopidy_uri(mopidy_uri)
|
|
|
|
return PodcastEpisode.find_or_create(**parsed_data)
|