mirrors/copyparty
1
0
Fork 0
mirror of https://github.com/9001/copyparty.git synced 2025-08-17 09:02:15 -06:00
Code Issues Actions Packages Projects Releases Wiki Activity

format with black

Browse source 
Signed-off-by: ed <s@ocv.me>
This commit is contained in:
ed 2025-08-10 20:29:20 +00:00 committed by GitHub
parent 988cccd2b8
commit 34bc52ecbd
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 22 additions and 17 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

39
copyparty/mtag.py
View file

@ -29,7 +29,7 @@ from .util import (
) )
if True: # pylint: disable=using-constant-test if True: # pylint: disable=using-constant-test
from typing import Any, Optional, Union, IO from typing import IO, Any, Optional, Union
from .util import NamedLogger, RootLogger from .util import NamedLogger, RootLogger
@ -60,6 +60,7 @@ def have_ff(scmd: str) -> bool:
else: else:
return bool(shutil.which(scmd)) return bool(shutil.which(scmd))
HAVE_FFMPEG = not os.environ.get("PRTY_NO_FFMPEG") and have_ff("ffmpeg") HAVE_FFMPEG = not os.environ.get("PRTY_NO_FFMPEG") and have_ff("ffmpeg")
HAVE_FFPROBE = not os.environ.get("PRTY_NO_FFPROBE") and have_ff("ffprobe") HAVE_FFPROBE = not os.environ.get("PRTY_NO_FFPROBE") and have_ff("ffprobe")
@ -369,7 +370,9 @@ def parse_ffprobe(txt: str) -> tuple[dict[str, tuple[int, Any]], dict[str, list[
def get_cover_from_epub(log: "NamedLogger", abspath: str) -> IO[bytes] | None: def get_cover_from_epub(log: "NamedLogger", abspath: str) -> IO[bytes] | None:
import zipfile import zipfile
from .dxml import parse_xml from .dxml import parse_xml
try: try:
from urlparse import urljoin # Python2 from urlparse import urljoin # Python2
except ImportError: except ImportError:
@ -380,33 +383,34 @@ def get_cover_from_epub(log: "NamedLogger", abspath: str) -> IO[bytes] | None:
try: try:
container_root = parse_xml(z.read("META-INF/container.xml").decode()) container_root = parse_xml(z.read("META-INF/container.xml").decode())
except KeyError: except KeyError:
log(f"epub: no container file found in {abspath}") log("epub: no container file found in %s" % (abspath,))
return None return None
# https://www.w3.org/TR/epub-33/#sec-container.xml-rootfile-elem # https://www.w3.org/TR/epub-33/#sec-container.xml-rootfile-elem
container_namespace = {"": "urn:oasis:names:tc:opendocument:xmlns:container"} container_ns = {"": "urn:oasis:names:tc:opendocument:xmlns:container"}
# One file could contain multiple package documents, default to the first one # One file could contain multiple package documents, default to the first one
rootfile_path = container_root\ rootfile_path = container_root.find("./rootfiles/rootfile", container_ns).get(
.find("./rootfiles/rootfile", container_namespace)\ "full-path"
.get("full-path") )
# Then open the first package document to find the path of the cover image # Then open the first package document to find the path of the cover image
try: try:
package_root = parse_xml(z.read(rootfile_path).decode()) package_root = parse_xml(z.read(rootfile_path).decode())
except KeyError: except KeyError:
log(f"epub: no package document found in {abspath}") log("epub: no package document found in %s" % (abspath,))
return None return None
# https://www.w3.org/TR/epub-33/#sec-package-doc # https://www.w3.org/TR/epub-33/#sec-package-doc
package_namespaces = {"": "http://www.idpf.org/2007/opf"} package_ns = {"": "http://www.idpf.org/2007/opf"}
# https://www.w3.org/TR/epub-33/#sec-cover-image # https://www.w3.org/TR/epub-33/#sec-cover-image
coverimage_path_node = package_root\ coverimage_path_node = package_root.find(
.find("./manifest/item[@properties='cover-image']", package_namespaces) "./manifest/item[@properties='cover-image']", package_ns
if coverimage_path_node: )
if coverimage_path_node is not None:
coverimage_path = coverimage_path_node.get("href") coverimage_path = coverimage_path_node.get("href")
else: else:
# This might be an EPUB2 file, try the legacy way of specifying covers # This might be an EPUB2 file, try the legacy way of specifying covers
coverimage_path = _get_cover_from_epub2(log, package_root, package_namespaces) coverimage_path = _get_cover_from_epub2(log, package_root, package_ns)
# This url is either absolute (in the .epub) or relative to the package document # This url is either absolute (in the .epub) or relative to the package document
adjusted_cover_path = urljoin(rootfile_path, coverimage_path) adjusted_cover_path = urljoin(rootfile_path, coverimage_path)
@ -414,16 +418,17 @@ def get_cover_from_epub(log: "NamedLogger", abspath: str) -> IO[bytes] | None:
return z.open(adjusted_cover_path) return z.open(adjusted_cover_path)
def _get_cover_from_epub2(log: "NamedLogger", package_root, package_namespaces) -> str | None: def _get_cover_from_epub2(log: "NamedLogger", package_root, package_ns) -> str | None:
# <meta name="cover" content="id-to-cover-image"> in <metadata>, then # <meta name="cover" content="id-to-cover-image"> in <metadata>, then
# <item> in <manifest> # <item> in <manifest>
cover_id = package_root \ cover_id = package_root.find("./metadata/meta[@name='cover']", package_ns).get(
.find("./metadata/meta[@name='cover']", package_namespaces) \ "content"
.get("content") )
if not cover_id: if not cover_id:
return None return None
for node in package_root.iterfind("./manifest/item", package_namespaces): for node in package_root.iterfind("./manifest/item", package_ns):
if node.get("id") == cover_id: if node.get("id") == cover_id:
cover_path = node.get("href") cover_path = node.get("href")
return cover_path return cover_path