f8bef94e7f
Sonarr (and potentially Radarr/Lidarr) can send the quality field as a plain string instead of the expected nested dict. Added a helper that handles both formats gracefully. https://claude.ai/code/session_019ANRdyL2jfi7ysWhqN4PfD
124 lines
3.5 KiB
Python
124 lines
3.5 KiB
Python
import markdown as md
|
|
|
|
|
|
def _render_html(text: str) -> str:
|
|
return md.markdown(text, extensions=["extra"])
|
|
|
|
|
|
def _strip_markdown(text: str) -> str:
|
|
return text.replace("**", "").replace("*", "").replace("> ", "")
|
|
|
|
|
|
def _extract_quality(quality_obj) -> str:
|
|
"""Extract quality name from *arr quality field, which may be a string or nested dict."""
|
|
if isinstance(quality_obj, str):
|
|
return quality_obj
|
|
if isinstance(quality_obj, dict):
|
|
inner = quality_obj.get("quality", quality_obj)
|
|
if isinstance(inner, str):
|
|
return inner
|
|
if isinstance(inner, dict):
|
|
return inner.get("name", "Unknown")
|
|
return "Unknown"
|
|
|
|
|
|
def format_radarr(payload: dict) -> tuple[str, str]:
|
|
movie = payload.get("movie", {})
|
|
title = movie.get("title", "Unknown")
|
|
year = movie.get("year", "")
|
|
is_upgrade = payload.get("isUpgrade", False)
|
|
|
|
quality = _extract_quality(
|
|
payload.get("movieFile", {}).get("quality", {})
|
|
)
|
|
|
|
if is_upgrade:
|
|
lines = [
|
|
f"\U0001f3ac **{title}** ({year})",
|
|
f"> \u2b06\ufe0f *Quality upgrade*",
|
|
f"> \U0001f39e\ufe0f Quality: {quality}",
|
|
]
|
|
else:
|
|
lines = [
|
|
f"\U0001f3ac **{title}** ({year})",
|
|
f"> \u2705 *New addition*",
|
|
f"> \U0001f39e\ufe0f Quality: {quality}",
|
|
]
|
|
|
|
body_md = "\n".join(lines)
|
|
plain = _strip_markdown(body_md)
|
|
html = _render_html(body_md)
|
|
return plain, html
|
|
|
|
|
|
def format_sonarr(payload: dict) -> tuple[str, str]:
|
|
series = payload.get("series", {})
|
|
series_title = series.get("title", "Unknown")
|
|
is_upgrade = payload.get("isUpgrade", False)
|
|
|
|
episodes = payload.get("episodes", [{}])
|
|
ep = episodes[0] if episodes else {}
|
|
season = ep.get("seasonNumber", 0)
|
|
episode = ep.get("episodeNumber", 0)
|
|
ep_title = ep.get("title", "")
|
|
|
|
quality = _extract_quality(
|
|
payload.get("episodeFile", {}).get("quality", {})
|
|
)
|
|
|
|
header = f"\U0001f4fa **{series_title}** \u2014 S{season:02d}E{episode:02d}"
|
|
if ep_title:
|
|
header += f" \u2014 *{ep_title}*"
|
|
|
|
if is_upgrade:
|
|
lines = [
|
|
header,
|
|
f"> \u2b06\ufe0f *Quality upgrade*",
|
|
f"> \U0001f39e\ufe0f Quality: {quality}",
|
|
]
|
|
else:
|
|
lines = [
|
|
header,
|
|
f"> \u2705 *New addition*",
|
|
f"> \U0001f39e\ufe0f Quality: {quality}",
|
|
]
|
|
|
|
body_md = "\n".join(lines)
|
|
plain = _strip_markdown(body_md)
|
|
html = _render_html(body_md)
|
|
return plain, html
|
|
|
|
|
|
def format_lidarr(payload: dict) -> tuple[str, str]:
|
|
artist = payload.get("artist", {})
|
|
artist_name = artist.get("name", "Unknown")
|
|
album = payload.get("album", {})
|
|
album_title = album.get("title", "")
|
|
is_upgrade = payload.get("isUpgrade", False)
|
|
|
|
track_files = payload.get("trackFiles", [{}])
|
|
tf = track_files[0] if track_files else {}
|
|
quality = _extract_quality(tf.get("quality", {}))
|
|
|
|
header = f"\U0001f3b5 **{artist_name}**"
|
|
if album_title:
|
|
header += f" \u2014 *{album_title}*"
|
|
|
|
if is_upgrade:
|
|
lines = [
|
|
header,
|
|
f"> \u2b06\ufe0f *Quality upgrade*",
|
|
f"> \U0001f3b5 Format: {quality}",
|
|
]
|
|
else:
|
|
lines = [
|
|
header,
|
|
f"> \u2705 *New addition*",
|
|
f"> \U0001f3b5 Format: {quality}",
|
|
]
|
|
|
|
body_md = "\n".join(lines)
|
|
plain = _strip_markdown(body_md)
|
|
html = _render_html(body_md)
|
|
return plain, html
|