mirror of
https://github.com/spaam/svtplay-dl.git
synced 2024-11-24 12:15:40 +01:00
51 lines
2.1 KiB
Python
51 lines
2.1 KiB
Python
import logging
|
|
|
|
import xml.etree.ElementTree as ET
|
|
from svtplay_dl.utils.output import formatname
|
|
from svtplay_dl.utils.parser import Options
|
|
from datetime import datetime
|
|
|
|
|
|
def write_nfo_episode(output, config):
|
|
root = ET.Element("episodedetails")
|
|
ET.SubElement(root, "title").text = output["title_nice"]
|
|
ET.SubElement(root, "showtitle").text = output["episodename"]
|
|
ET.SubElement(root, "season").text = output["season"]
|
|
ET.SubElement(root, "episode").text = output["episode"]
|
|
ET.SubElement(root, "plot").text = output["showdescription"]
|
|
if output["publishing_datetime"] is not None:
|
|
ET.SubElement(root, "aired").text = datetime.fromtimestamp(output["publishing_datetime"]).isoformat()
|
|
if not config.get("thumbnail"):
|
|
# Set the thumbnail path to download link if not thumbnail downloaded
|
|
ET.SubElement(root, "thumb").text = output["showthumbnailurl"]
|
|
|
|
filename = formatname(output.copy(), config, extension="nfo")
|
|
logging.info("NFO episode: %s", filename)
|
|
|
|
tree = ET.ElementTree(root)
|
|
tree.write(filename, encoding="UTF-8", xml_declaration=True)
|
|
|
|
|
|
def write_nfo_tvshow(output, config):
|
|
# Config for tvshow nfo file
|
|
root = ET.Element("tvshow")
|
|
ET.SubElement(root, "title").text = output["title_nice"]
|
|
ET.SubElement(root, "plot").text = output["episodedescription"]
|
|
if config.get("thumbnail"):
|
|
# Set the thumbnail relative path to downloaded thumbnail
|
|
ET.SubElement(root, "thumb").text = "{}.tvshow.tbn".format(output["title"])
|
|
else:
|
|
# Set the thumbnail path to download link if not thumbnail downloaded
|
|
ET.SubElement(root, "thumb").text = output["episodethumbnailurl"]
|
|
|
|
cconfig = Options()
|
|
cconfig.set("output", config.get("output"))
|
|
cconfig.set("path", config.get("path"))
|
|
cconfig.set("subfolder", config.get("subfolder"))
|
|
cconfig.set("filename", "tvshow.{ext}")
|
|
filename = formatname(output.copy(), cconfig, extension="nfo")
|
|
logging.info("NFO show: %s", filename)
|
|
|
|
tree = ET.ElementTree(root)
|
|
tree.write(filename, encoding="UTF-8", xml_declaration=True)
|