mirror of
https://github.com/ZuneDev/PyZuneCatalogServer.git
synced 2026-07-27 13:11:13 -07:00
76 lines
2.3 KiB
Python
76 lines
2.3 KiB
Python
|
from typing import List
|
||
|
|
from xml.dom import minidom
|
||
|
|
from xml.dom.minidom import Element, Document, Text
|
||
|
|
from datetime import datetime
|
||
|
|
|
||
|
|
MIME_XML = "text/xml"
|
||
|
|
MIME_ATOM_XML = "application/atom+xml"
|
||
|
|
MIME_UIX = "application/uix"
|
||
|
|
MIME_JPG = "image/jpeg"
|
||
|
|
|
||
|
|
|
||
|
|
def set_element_value(element: Element, value: str):
|
||
|
|
content = Text()
|
||
|
|
content.data = value
|
||
|
|
element.appendChild(content)
|
||
|
|
|
||
|
|
|
||
|
|
def create_feed(doc: Document, title: str, id: str, href: str, date_updated: datetime = datetime.today()) -> Element:
|
||
|
|
feed = create_empty_feed(doc)
|
||
|
|
|
||
|
|
feed.appendChild(create_link(doc, href))
|
||
|
|
feed.appendChild(create_updated(doc, date_updated))
|
||
|
|
feed.appendChild(create_title(doc, title))
|
||
|
|
feed.appendChild(create_id(doc, id))
|
||
|
|
|
||
|
|
return feed
|
||
|
|
|
||
|
|
|
||
|
|
def create_empty_feed(doc: Document) -> Element:
|
||
|
|
# Add namespaces
|
||
|
|
feed: Element = doc.createElement("a:feed")
|
||
|
|
feed.setAttribute("xmlns:a", "http://www.w3.org/2005/Atom")
|
||
|
|
feed.setAttribute("xmlns:os", "http://a9.com/-/spec/opensearch/1.1/")
|
||
|
|
feed.setAttribute("xmlns", "http://schemas.zune.net/catalog/music/2007/10")
|
||
|
|
doc.appendChild(feed)
|
||
|
|
|
||
|
|
return feed
|
||
|
|
|
||
|
|
|
||
|
|
def create_link(doc: Document, href: str, rel: str = "self", type: str = MIME_ATOM_XML) -> Element:
|
||
|
|
link: Element = doc.createElement("a:link")
|
||
|
|
link.setAttribute("rel", rel)
|
||
|
|
link.setAttribute("type", type)
|
||
|
|
link.setAttribute("href", href)
|
||
|
|
return link
|
||
|
|
|
||
|
|
|
||
|
|
def create_updated(doc: Document, date_updated: datetime = datetime.today()) -> Element:
|
||
|
|
updated: Element = doc.createElement("a:updated")
|
||
|
|
set_element_value(updated, date_updated.isoformat())
|
||
|
|
return updated
|
||
|
|
|
||
|
|
|
||
|
|
def create_title(doc: Document, title: str, type: str = "text") -> Element:
|
||
|
|
title_elem: Element = doc.createElement("a:title")
|
||
|
|
title_elem.setAttribute("type", type)
|
||
|
|
set_element_value(title_elem, title)
|
||
|
|
return title_elem
|
||
|
|
|
||
|
|
|
||
|
|
def create_id(doc: Document, id: str) -> Element:
|
||
|
|
id_elem: Element = doc.createElement("a:id")
|
||
|
|
set_element_value(id_elem, id)
|
||
|
|
return id_elem
|
||
|
|
|
||
|
|
|
||
|
|
def create_entry(doc: Document, title: str, id: str, href: str, date_updated: datetime = datetime.today()) -> Element:
|
||
|
|
entry: Element = doc.createElement("a:entry")
|
||
|
|
|
||
|
|
entry.appendChild(create_link(doc, href))
|
||
|
|
entry.appendChild(create_updated(doc, date_updated))
|
||
|
|
entry.appendChild(create_title(doc, title))
|
||
|
|
entry.appendChild(create_id(doc, id))
|
||
|
|
|
||
|
|
return entry
|