refactor: Refactor mpegdash patch logic and update import ordering
This commit is contained in:
@@ -1,10 +1,4 @@
|
|||||||
#!/usr/bin/env python
|
#!/usr/bin/env python
|
||||||
# Apply mpegdash patch before any tidalapi imports to fix TIDAL manifest parsing
|
|
||||||
# See: https://github.com/FunWarry/tidal-dl-ng-For-DJ/issues/15
|
|
||||||
from tidal_dl_ng.helper.mpegdash_patch import apply_mpegdash_patch
|
|
||||||
|
|
||||||
apply_mpegdash_patch()
|
|
||||||
|
|
||||||
import importlib.metadata
|
import importlib.metadata
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
from urllib.parse import urlparse
|
from urllib.parse import urlparse
|
||||||
@@ -13,8 +7,14 @@ import requests
|
|||||||
import toml
|
import toml
|
||||||
|
|
||||||
from tidal_dl_ng.constants import REQUESTS_TIMEOUT_SEC
|
from tidal_dl_ng.constants import REQUESTS_TIMEOUT_SEC
|
||||||
|
|
||||||
|
# Apply mpegdash patch before any tidalapi imports to fix TIDAL manifest parsing
|
||||||
|
# See: https://github.com/FunWarry/tidal-dl-ng-For-DJ/issues/15
|
||||||
|
from tidal_dl_ng.helper.mpegdash_patch import apply_mpegdash_patch
|
||||||
from tidal_dl_ng.model.meta import ProjectInformation, ReleaseLatest
|
from tidal_dl_ng.model.meta import ProjectInformation, ReleaseLatest
|
||||||
|
|
||||||
|
apply_mpegdash_patch()
|
||||||
|
|
||||||
|
|
||||||
def metadata_project() -> ProjectInformation:
|
def metadata_project() -> ProjectInformation:
|
||||||
result: ProjectInformation
|
result: ProjectInformation
|
||||||
|
|||||||
@@ -10,6 +10,7 @@ non-integer values for attributes that are expected to be integers.
|
|||||||
"""
|
"""
|
||||||
|
|
||||||
import logging
|
import logging
|
||||||
|
import re
|
||||||
|
|
||||||
logger = logging.getLogger(__name__)
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
@@ -24,6 +25,36 @@ def _safe_int(value: str) -> int | None:
|
|||||||
return None
|
return None
|
||||||
|
|
||||||
|
|
||||||
|
def _convert_list(attr_val: str, value_type) -> list[str]:
|
||||||
|
attr_type = value_type[0] if value_type else str
|
||||||
|
try:
|
||||||
|
return [attr_type(elem) for elem in re.split(r"[, ]", attr_val)]
|
||||||
|
except (ValueError, TypeError):
|
||||||
|
return [str(elem) for elem in re.split(r"[, ]", attr_val)]
|
||||||
|
|
||||||
|
|
||||||
|
def _convert_single(attr_name: str, attr_val: str, value_type):
|
||||||
|
if value_type == int:
|
||||||
|
result = _safe_int(attr_val)
|
||||||
|
if result is None:
|
||||||
|
logger.debug(
|
||||||
|
"mpegdash: Could not convert '%s'='%s' to int, using None",
|
||||||
|
attr_name,
|
||||||
|
attr_val,
|
||||||
|
)
|
||||||
|
return result
|
||||||
|
try:
|
||||||
|
return value_type(attr_val)
|
||||||
|
except (ValueError, TypeError):
|
||||||
|
logger.debug(
|
||||||
|
"mpegdash: Could not convert '%s'='%s' to %s, using None",
|
||||||
|
attr_name,
|
||||||
|
attr_val,
|
||||||
|
value_type.__name__,
|
||||||
|
)
|
||||||
|
return None
|
||||||
|
|
||||||
|
|
||||||
def apply_mpegdash_patch() -> None:
|
def apply_mpegdash_patch() -> None:
|
||||||
"""
|
"""
|
||||||
Apply a monkey-patch to mpegdash to handle string values in integer fields.
|
Apply a monkey-patch to mpegdash to handle string values in integer fields.
|
||||||
@@ -38,46 +69,18 @@ def apply_mpegdash_patch() -> None:
|
|||||||
|
|
||||||
try:
|
try:
|
||||||
from mpegdash import utils as mpegdash_utils
|
from mpegdash import utils as mpegdash_utils
|
||||||
import re
|
|
||||||
|
|
||||||
# Store the original function
|
|
||||||
original_parse_attr_value = mpegdash_utils.parse_attr_value
|
|
||||||
|
|
||||||
def patched_parse_attr_value(xmlnode, attr_name, value_type):
|
def patched_parse_attr_value(xmlnode, attr_name, value_type):
|
||||||
"""
|
if attr_name not in xmlnode.attributes:
|
||||||
Patched version of parse_attr_value that handles non-integer values gracefully.
|
|
||||||
"""
|
|
||||||
if attr_name not in xmlnode.attributes.keys():
|
|
||||||
return None
|
return None
|
||||||
|
|
||||||
attr_val = xmlnode.attributes[attr_name].nodeValue
|
attr_val = xmlnode.attributes[attr_name].nodeValue
|
||||||
|
|
||||||
if isinstance(value_type, list):
|
if isinstance(value_type, list):
|
||||||
attr_type = value_type[0] if len(value_type) > 0 else str
|
return _convert_list(attr_val, value_type)
|
||||||
try:
|
|
||||||
return [attr_type(elem) for elem in re.split(r"[, ]", attr_val)]
|
|
||||||
except (ValueError, TypeError):
|
|
||||||
# If conversion fails, return as strings
|
|
||||||
return [str(elem) for elem in re.split(r"[, ]", attr_val)]
|
|
||||||
|
|
||||||
# Handle integer conversion failures gracefully
|
return _convert_single(attr_name, attr_val, value_type)
|
||||||
if value_type == int:
|
|
||||||
result = _safe_int(attr_val)
|
|
||||||
if result is None:
|
|
||||||
logger.debug(
|
|
||||||
f"mpegdash: Could not convert '{attr_name}'='{attr_val}' to int, using None"
|
|
||||||
)
|
|
||||||
return result
|
|
||||||
|
|
||||||
try:
|
|
||||||
return value_type(attr_val)
|
|
||||||
except (ValueError, TypeError):
|
|
||||||
logger.debug(
|
|
||||||
f"mpegdash: Could not convert '{attr_name}'='{attr_val}' to {value_type.__name__}, using None"
|
|
||||||
)
|
|
||||||
return None
|
|
||||||
|
|
||||||
# Apply the patch
|
|
||||||
mpegdash_utils.parse_attr_value = patched_parse_attr_value
|
mpegdash_utils.parse_attr_value = patched_parse_attr_value
|
||||||
_patched = True
|
_patched = True
|
||||||
logger.debug("mpegdash patch applied successfully")
|
logger.debug("mpegdash patch applied successfully")
|
||||||
@@ -85,4 +88,4 @@ def apply_mpegdash_patch() -> None:
|
|||||||
except ImportError:
|
except ImportError:
|
||||||
logger.warning("Could not import mpegdash, patch not applied")
|
logger.warning("Could not import mpegdash, patch not applied")
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
logger.warning(f"Failed to apply mpegdash patch: {e}")
|
logger.warning("Failed to apply mpegdash patch: %s", e)
|
||||||
|
|||||||
Reference in New Issue
Block a user