pyrekordbox 0.3.1__py3-none-any.whl → 0.4.0__py3-none-any.whl

pyrekordbox/db6/smart_playlist.py

@@ -1,333 +0,0 @@
-# -*- coding: utf-8 -*-
-# Author: Dylan Jones
-# Date:   2023-12-13
-
-import xml.etree.cElementTree as xml
-from enum import IntEnum
-from typing import List, Union
-from datetime import datetime
-from dataclasses import dataclass
-
-from sqlalchemy import or_, and_, not_
-from sqlalchemy.sql.elements import BooleanClauseList
-from dateutil.relativedelta import relativedelta  # noqa
-
-from .tables import DjmdContent
-
-__all__ = [
-    "LogicalOperator",
-    "Operator",
-    "Condition",
-    "SmartList",
-]
-
-
-class LogicalOperator(IntEnum):
-    ALL = 1
-    ANY = 2
-
-
-class Operator(IntEnum):
-    EQUAL = 1
-    NOT_EQUAL = 2
-    GREATER = 3
-    LESS = 4
-    IN_RANGE = 5
-    IN_LAST = 6
-    NOT_IN_LAST = 7
-    CONTAINS = 8
-    NOT_CONTAINS = 9
-    STARTS_WITH = 10
-    ENDS_WITH = 11
-
-
-STR_OPS = [
-    Operator.EQUAL,
-    Operator.NOT_EQUAL,
-    Operator.CONTAINS,
-    Operator.NOT_CONTAINS,
-    Operator.STARTS_WITH,
-    Operator.ENDS_WITH,
-]
-
-NUM_OPS = [
-    Operator.EQUAL,
-    Operator.NOT_EQUAL,
-    Operator.GREATER,
-    Operator.LESS,
-    Operator.IN_RANGE,
-]
-
-DATE_OPS = [
-    Operator.EQUAL,
-    Operator.NOT_EQUAL,
-    Operator.GREATER,
-    Operator.LESS,
-    Operator.IN_RANGE,
-    Operator.IN_LAST,
-    Operator.NOT_IN_LAST,
-]
-
-
-PROPERTY_MAP = {
-    "artist": "ArtistName",
-    "album": "AlbumName",
-    "albumArtist": "AlbumArtist",
-    "originalArtist": "OrgArtist",
-    "bpm": "BPM",
-    "grouping": "ColorID",
-    "comments": "Commnt",
-    "producer": "ComposerName",
-    "stockDate": "StockDate",
-    "dateCreated": "created_at",
-    "counter": "DJPlayCount",
-    "fileName": "FileNameL",
-    "genre": "GenreName",
-    "key": "Key",
-    "label": "LabelName",
-    "mixName": "",
-    "myTag": "myTag",
-    "rating": "Rating",
-    "dateReleased": "ReleaseDate",
-    "remixedBy": "RemixerName",
-    "duration": "Length",
-    "name": "Title",
-    "year": "ReleaseYear",
-}
-
-
-VALID_OPS = {
-    "artist": STR_OPS,
-    "album": STR_OPS,
-    "albumArtist": STR_OPS,
-    "originalArtist": STR_OPS,
-    "bpm": NUM_OPS,
-    "grouping": [Operator.EQUAL, Operator.NOT_EQUAL],
-    "comments": STR_OPS,
-    "producer": STR_OPS,
-    "stockDate": DATE_OPS,
-    "dateCreated": DATE_OPS,
-    "counter": NUM_OPS,
-    "fileName": STR_OPS,
-    "genre": STR_OPS,
-    "key": STR_OPS,
-    "label": STR_OPS,
-    "mixName": STR_OPS,
-    "myTag": [Operator.CONTAINS, Operator.NOT_CONTAINS],
-    "rating": NUM_OPS,
-    "dateReleased": DATE_OPS,
-    "remixedBy": STR_OPS,
-    "duration": NUM_OPS,
-    "name": STR_OPS,
-    "year": NUM_OPS,
-}
-
-TYPE_CONVERSION = {
-    "bpm": int,
-    "stockDate": lambda x: datetime.strptime(x, "%Y-%m-%d"),
-    "dateCreated": lambda x: datetime.strptime(x, "%Y-%m-%d"),
-    "counter": int,
-    "rating": int,
-    "dateReleased": lambda x: datetime.strptime(x, "%Y-%m-%d"),
-    "duration": int,
-    "year": int,
-}
-
-
-@dataclass
-class Condition:
-    """Dataclass for a smart playlist condition."""
-
-    property: str
-    operator: int
-    unit: str
-    value_left: Union[str, int]
-    value_right: Union[str, int]
-
-    def __post_init__(self):
-        if self.property not in VALID_OPS:
-            raise ValueError(f"Invalid property: '{self.property}'")
-
-        valid_ops = VALID_OPS[self.property]
-        if self.operator not in valid_ops:
-            raise ValueError(
-                f"Invalid operator '{self.operator}' for '{self.property}', "
-                f"must be one of {valid_ops}"
-            )
-
-        if self.operator == Operator.IN_RANGE:
-            if not self.value_right:
-                raise ValueError(f"Operator '{self.operator}' requires `value_right`")
-
-
-def _get_condition_values(cond):
-    val_left = cond.value_left
-    val_right = cond.value_right
-    func = None
-    if cond.operator in (Operator.IN_LAST, Operator.NOT_IN_LAST):
-        func = int
-    elif cond.property in TYPE_CONVERSION:
-        func = TYPE_CONVERSION[cond.property]
-
-    if func is not None:
-        if val_left != "":
-            val_left = func(val_left)
-        if val_right != "":
-            try:
-                val_right = func(val_right)
-            except ValueError:
-                pass
-    return val_left, val_right
-
-
-class SmartList:
-    """Rekordbox smart playlist XML handler."""
-
-    def __init__(
-        self,
-        playlist_id: Union[int, str] = None,
-        logical_operator: int = None,
-        auto_update: int = 1,
-    ):
-        self.playlist_id: Union[int, str] = playlist_id
-        self.logical_operator: int = logical_operator
-        self.auto_update: int = auto_update
-        self.conditions: List[Condition] = list()
-
-    def parse(self, source: str) -> None:
-        """Parse the XML source of a smart playlist."""
-        tree = xml.ElementTree(xml.fromstring(source))
-        root = tree.getroot()
-        conditions = list()
-        for child in root.findall("CONDITION"):
-            condition = Condition(
-                property=child.attrib["PropertyName"],
-                operator=int(child.attrib["Operator"]),
-                unit=child.attrib["ValueUnit"],
-                value_left=child.attrib["ValueLeft"],
-                value_right=child.attrib["ValueRight"],
-            )
-            conditions.append(condition)
-
-        self.playlist_id = int(root.attrib["Id"])
-        self.logical_operator = int(root.attrib["LogicalOperator"])
-        self.auto_update = int(root.attrib["AutomaticUpdate"])
-        self.conditions = conditions
-
-    def to_xml(self) -> str:
-        """Convert the smart playlist conditions to XML."""
-        attrib = {
-            "Id": str(self.playlist_id),
-            "LogicalOperator": str(self.logical_operator),
-            "AutomaticUpdate": str(self.auto_update),
-        }
-        root = xml.Element("NODE", attrib=attrib)
-        for cond in self.conditions:
-            attrib = {
-                "PropertyName": str(cond.property),
-                "Operator": str(cond.operator),
-                "ValueUnit": str(cond.unit),
-                "ValueLeft": str(cond.value_left),
-                "ValueRight": str(cond.value_right),
-            }
-            xml.SubElement(root, "CONDITION", attrib=attrib)
-        return xml.tostring(root).decode("utf-8").replace(" /", "/")
-
-    def add_condition(
-        self,
-        prop: str,
-        operator: int,
-        value_left: str,
-        value_right: str = "",
-        unit: str = "",
-    ) -> None:
-        """Add a condition to the smart playlist.
-
-        Parameters
-        ----------
-        prop : str
-            The property to filter on.
-        operator : {1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11} int
-            The operator to use. Must be in the range of 1-11
-        value_left : str
-            The left value to use.
-        value_right : str, optional
-            The right value to use, by default "".
-        unit : str, optional
-            The unit to use, by default "".
-        """
-        cond = Condition(prop, operator, unit, value_left, value_right)
-        self.conditions.append(cond)
-
-    def filter_clause(self) -> BooleanClauseList:
-        """Return a SQLAlchemy filter clause matching the content of the smart playlist.
-
-        Returns
-        -------
-        BooleanClauseList
-            A filter list macthing the contents of the smart playlist.
-        """
-        conditions = self.conditions
-        logical_op = and_ if self.logical_operator == LogicalOperator.ALL else or_
-        cond = conditions[0]
-
-        colum_name = PROPERTY_MAP[cond.property]
-
-        comps = list()
-        for cond in conditions:
-            val_left, val_right = _get_condition_values(cond)
-
-            if cond.operator == Operator.EQUAL:
-                comp = getattr(DjmdContent, colum_name) == val_left
-
-            elif cond.operator == Operator.NOT_EQUAL:
-                comp = getattr(DjmdContent, colum_name) != val_left
-
-            elif cond.operator == Operator.GREATER:
-                comp = getattr(DjmdContent, colum_name) > val_left
-
-            elif cond.operator == Operator.LESS:
-                comp = getattr(DjmdContent, colum_name) < val_left
-
-            elif cond.operator == Operator.IN_RANGE:
-                comp = getattr(DjmdContent, colum_name).between(val_left, val_right)
-
-            elif cond.operator == Operator.CONTAINS:
-                comp = getattr(DjmdContent, colum_name).contains(val_left)
-
-            elif cond.operator == Operator.NOT_CONTAINS:
-                comp = not_(getattr(DjmdContent, colum_name).contains(val_left))
-
-            elif cond.operator == Operator.STARTS_WITH:
-                comp = getattr(DjmdContent, colum_name).startswith(val_left)
-
-            elif cond.operator == Operator.ENDS_WITH:
-                comp = getattr(DjmdContent, colum_name).endswith(val_left)
-
-            elif cond.operator == Operator.IN_LAST:
-                now = datetime.now()
-                if cond.unit == "day":
-                    t0 = now - relativedelta(days=val_left)
-                    comp = getattr(DjmdContent, colum_name) > t0
-                elif cond.unit == "month":
-                    t0 = now - relativedelta(months=val_left)
-                    comp = getattr(DjmdContent, colum_name).month > t0
-                else:
-                    raise ValueError(f"Unknown unit '{cond.unit}'")
-
-            elif cond.operator == Operator.NOT_IN_LAST:
-                now = datetime.now()
-                if cond.unit == "day":
-                    t0 = now - relativedelta(days=val_left)
-                    comp = getattr(DjmdContent, colum_name) < t0
-                elif cond.unit == "month":
-                    t0 = now - relativedelta(months=val_left)
-                    comp = getattr(DjmdContent, colum_name).month < t0
-                else:
-                    raise ValueError(f"Unknown unit '{cond.unit}'")
-
-            else:
-                raise ValueError(f"Unknown operator '{cond.operator}'")
-
-            comps.append(comp)
-        return logical_op(*comps)

pyrekordbox/xml.py

@@ -1,8 +0,0 @@
-# -*- coding: utf-8 -*-
-# Author: Dylan Jones
-# Date:   2022-04-10
-
-from .utils import warn_deprecated as _warn_deprecated
-from .rbxml import *  # noqa: F401, F403
-
-_warn_deprecated("pyrekordbox.xml", "pyrekordbox.rbxml", remove_in="0.4.0")

pyrekordbox-0.3.1.dist-info/RECORD

@@ -1,84 +0,0 @@
-docs/Makefile,sha256=4zv3TVkTACm6JBaKgTES3ZI9cETXgM6ULbZkXZP1as8,638
-docs/make.bat,sha256=s8EuuVXNRnn4xmWLWTpk3Z01aqJSJT8ymrmK6ux0zbc,769
-docs/source/api.md,sha256=IKKMeQ2wFSS_ippOWJ63ELtMjfkNe27KjWcFFEdUXnE,276
-docs/source/conf.py,sha256=lPfAJtd0VY7uEDmHrEfHvQzNJg6WN_1seM9pCFc2e6E,6244
-docs/source/index.md,sha256=c67TMpDCaiqFe44n0pXWL__utlLAz7xqfDVjyWbxx3M,2660
-docs/source/installation.md,sha256=WmX6gjuPimA6uXhS7wZzKKG1AQCHpfB0nPm3g8_ur6A,9234
-docs/source/key.md,sha256=J9kx2Cgwm-ddLjbFkjBmAargPSg5WfVguJ1ffrDuCDs,3976
-docs/source/quickstart.md,sha256=kKau0nQHc0RByrVq0pgaDKWTUX_l4290ruqY08mXTCg,6301
-docs/source/requirements.txt,sha256=7YYc-tO0BQcBVUa4GvM1r1G5Ma9gkwGZpNbAKlJL97s,117
-docs/source/_static/images/anlz_beat.svg,sha256=cVooi17HFw3mOT39cFNB05tr-cZwbK_Hjmho6eBJNX8,2924
-docs/source/_static/images/anlz_file.svg,sha256=E74hjyTFJRy6tCxyTPfsYoM0GmOKoY1To_eHatpPEjs,12320
-docs/source/_static/images/anlz_pco2.svg,sha256=LAv_EZf72FbslKdHmuBy5ROWy8xKKkkXqp7mSk3WG20,8155
-docs/source/_static/images/anlz_pcob.svg,sha256=w4NRLCj0EMMnQ2i6tnlE-nFxETSiTs8PQF5nxMI3bgo,8682
-docs/source/_static/images/anlz_pcp2.svg,sha256=JDncv7fo6ylorWyaH7kZ6roYB7zryFbJE_HY4IfWG8E,24904
-docs/source/_static/images/anlz_pcpt.svg,sha256=8PcHBFAHXrDnTyC_7rO0TM-5cGy8BfjIXWxDi0zEK8Y,16473
-docs/source/_static/images/anlz_ppth.svg,sha256=-rPlwvBrrsYaL87GNI-WJJBFCYkcZwkWoGwH5NT8bLk,7124
-docs/source/_static/images/anlz_pqt2.svg,sha256=sRvbV-DjoTIWtxwChaw6kBqKFXNqBCF4V9hru9ZtdiE,18386
-docs/source/_static/images/anlz_pqt2_2.svg,sha256=8g3XCJDz8xbH_KK7CvMpmIZjaGrWBKQVlA4k4M3PF50,13774
-docs/source/_static/images/anlz_pqtz.svg,sha256=OPlW76hPkbhOByYgmVQfucTpb5irZXYcASZokfpK3A0,8141
-docs/source/_static/images/anlz_pssi.svg,sha256=zM8fsNshSMd0wDvO_qM7RzsAKVIEN8Y41jY76hBJjyQ,11684
-docs/source/_static/images/anlz_pssi_entry.svg,sha256=YkUg7fXtUst0OBI6vIlhHCK-nzhmR-22UGKFQ0pHxxA,12095
-docs/source/_static/images/anlz_pvbr.svg,sha256=7mA6S__F4AoX-GamftlevellDiTlcIWRWi77A7HZEDg,7284
-docs/source/_static/images/anlz_pwav.svg,sha256=psm3g7ILLVNgPNmPG2Hy7k9-q7qmnMnyQ3_-XYVAW2g,7566
-docs/source/_static/images/anlz_pwv3.svg,sha256=FwZJIiHj5hj-FB-cioNlxrGBeIEuZy2BOLEzSWC57IE,8389
-docs/source/_static/images/anlz_pwv4.svg,sha256=k5GQu2N965zPSG0FQWuml96UMfqBvs8Vy-Emklvn2dI,8173
-docs/source/_static/images/anlz_pwv5.svg,sha256=Ph1TSlji2BgWK3rVI7MvnpGM68wef1s-Hs6OlmC0oQQ,8173
-docs/source/_static/images/anlz_pwv5_entry.svg,sha256=TyNh_6gIsQKuDvS0RO5tmsptzmxiQ-LL1L3DQdIFi4I,5648
-docs/source/_static/images/anlz_pwv6.svg,sha256=J2KvE_bI-NTFSpFp7QsKSsgbWGCyuUAuW_dOgdnv2DY,7577
-docs/source/_static/images/anlz_pwv7.svg,sha256=n5NsTErvfsCw-otWdVXn12lM0tHyh-zQDZsZtQOJf0U,8173
-docs/source/_static/images/anlz_pwvc.svg,sha256=wO_tsxRDVnAbptbSZYK1z7YOH-0vNUqXDB9Udj4ungA,7128
-docs/source/_static/images/anlz_tag.svg,sha256=owPTtrTRDYCkaSg32_UJ-MTZp3K6WlkRQdJHTzdtxqs,6188
-docs/source/_static/images/x64dbg_rb_key.png,sha256=gtwUA6lbuFRbLQQACnBQ-aO8p5Ex-7E41dS0cW6XyAU,280011
-docs/source/_static/logos/dark/logo_primary.svg,sha256=pH4Q28u9JaCA4dVrKB-NDy00PNw2EhDwz7fIYUzezvk,3079
-docs/source/_static/logos/light/logo_primary.svg,sha256=MUbYjowOcWdtoHVnK7pyV9xcTRdm7gS4whw94pVMGSI,3080
-docs/source/_static/logos/mid/logo_primary.svg,sha256=utCWWbXXW5auRRposMMaMO9t_AflsN13rZGo3_Pg2WE,3078
-docs/source/_templates/apidoc/module.rst_t,sha256=keQqb9G8LA-JS5tuytgzfsxRXsADEvuOz3mJjY9qYHU,185
-docs/source/_templates/apidoc/package.rst_t,sha256=Ra-lunZcMyDtrb6vgrfAnBcBW4eFMTEIyg12fSe9h4o,1173
-docs/source/_templates/apidoc/toc.rst_t,sha256=6EfpNwcRXLa8Ae3NcovuSmV5PSBy5gP0t6XfphUYb7M,127
-docs/source/_templates/autosummary/class.rst,sha256=5yRMWkUNDXGvQE1gGm6yisL1bX4BDOsaBPciNUs_6vA,579
-docs/source/_templates/autosummary/module.rst,sha256=xSCcWeBqQdMgpese2OwFlvzSOMsPZKTBCLCwgD3OG4k,789
-docs/source/development/changes.md,sha256=ys1_zvYKQWEak_mBXBUMLpO3CWriHpjSAg1-17mTk04,40
-docs/source/development/contributing.md,sha256=7bERN03kHo2Wf0ZrgHYP4MfBSdFGmE_Kk0CxK0uay2I,43
-docs/source/formats/anlz.md,sha256=P7FrPqbu3i8SlRKVV9IU9cOQajKJHOjLlAWHheDGN7c,21455
-docs/source/formats/db6.md,sha256=fuB1_PaalHV4zRkhXh5_mn1ZgHfPnSu_OzUOYbVls6w,30121
-docs/source/formats/mysetting.md,sha256=OIYIn_B2HJcJtJM0wNjjzPy3xPkyz4raxg_-Y6E3D1k,9746
-docs/source/formats/xml.md,sha256=uL7eWWQfQXsyf55Vydt1iLq7Jzio3kIGbSfjpg5-CRY,8491
-docs/source/tutorial/anlz.md,sha256=JtcQIy44qaSP_GL0aL9VYQYOX8aVju7uFxgN3FPNy9o,145
-docs/source/tutorial/configuration.md,sha256=mqmeFM3I7pDA_ajPQWqVyvc9Y0PWGYXLr3yyojTGsL8,2127
-docs/source/tutorial/db6.md,sha256=nmaC6sF1noBxp0_LvihnzD9eMYeL-fCkAet4qiShOcc,6611
-docs/source/tutorial/index.md,sha256=zjx5uXESYzHIyeqtL4XjhZUDPOpNiutZ3gkUOeG8R9w,318
-docs/source/tutorial/mysetting.md,sha256=su-WMvEd3DrFF32RDfQBm3TlAUgK5PRPT1n_77BYnvY,3659
-docs/source/tutorial/xml.md,sha256=x_rFvsVzZtUV0upEkPWkgrmr3B-FYjZKWvBuJBLFAu0,6414
-pyrekordbox/__init__.py,sha256=ODYrBuSF14PNKNAxsbi5_VR-ja2P3ySK0WwM8l7MFCk,647
-pyrekordbox/__main__.py,sha256=R1lus_-hBG4Uf6SRuzXXFwaE-dT9Uma_lzMOGgsdDLs,5975
-pyrekordbox/_version.py,sha256=HzPz9rq3s1AiZXregKlqKaJJ2wGMtvH_a3V9la9CnpM,411
-pyrekordbox/config.py,sha256=ttihHzwfg3IGnp2IyCdnOWKhiH72HZ-dy7NQb4-SG-8,25190
-pyrekordbox/logger.py,sha256=qCY_3L_3WIMAvNmVKvh7oshGlWrd0T5aIW1bNCIu3Lo,521
-pyrekordbox/rbxml.py,sha256=0qsKc8SQa54URwWGt6oHTOHBiAQU2S8MbSxDbzKUB7A,38369
-pyrekordbox/utils.py,sha256=sT1xt-rM94Dir-S0OSSqKbquf9kdFbzZvBu3-7q0QDM,4361
-pyrekordbox/xml.py,sha256=7wFkMyWL4AQLINvxVxkQxlruG9c7R-gkxdP7xXFOkWU,241
-pyrekordbox/anlz/__init__.py,sha256=OSpyl3pmmnt3mG905J9qvNnYchsgrIfathw0QXoOhjo,3156
-pyrekordbox/anlz/file.py,sha256=r2FppO6efPzKfCfz6foX9G2aBu8gkKXMrjIm9fua3_M,6949
-pyrekordbox/anlz/structs.py,sha256=Lt4fkb3SAE8w146eWeWGnpgRoP6jhLMWrSMoMwPjG04,7925
-pyrekordbox/anlz/tags.py,sha256=uqneBP9CstYy7a4IpeEC9A5tfI59DQm0ik7yFXk_dNo,14094
-pyrekordbox/db6/__init__.py,sha256=d2ITtFzw41QfaI_v6QPYpr2KxgziGdsVyxOdXTvlPH0,886
-pyrekordbox/db6/aux_files.py,sha256=t0SfMCXBHjlCSfgRfpw1RTj0cmV0QY07xSLepjQqMnY,7591
-pyrekordbox/db6/database.py,sha256=dO2v7JNBMLDt7g22vLDmegzP3TYeeYiBnQqE5P5q8AE,78252
-pyrekordbox/db6/registry.py,sha256=L4X49HqKtvyD_c4Ad3HR7aM5bj468MgD3Yj6WUE0ri8,9724
-pyrekordbox/db6/smart_playlist.py,sha256=BOZrPHUU1r0_dM3fwLn8flBi5tigeScENoB3RFoyHiY,9911
-pyrekordbox/db6/tables.py,sha256=EnfZscPVVeVKmVTFmDoXT3sHMcfjgtq8j-GIMpElMuU,66326
-pyrekordbox/mysettings/__init__.py,sha256=rMS6Kknf1-X3PXF_TUxm8xui0H1Ap3R7G_9mqlnQUAM,705
-pyrekordbox/mysettings/file.py,sha256=dQEsdBivpverXCssyVNUeQyUXIooNrOGIIiPLwi_6T4,12669
-pyrekordbox/mysettings/structs.py,sha256=5Y1F3qTmsP1fRB39_BEHpQVxKx2DO9BytEuJUG_RNcY,8472
-tests/__init__.py,sha256=s321RtRiHJsLt0YFq0NRncW7u66uwYHQE-6IjPcg10o,67
-tests/test_anlz.py,sha256=hZAW5GFEJpU8UmmHYLuU5OZlApGrWZyPAR79KtaR0cw,5782
-tests/test_config.py,sha256=wYp4wHRhIk2grhGXhk4z9umrhOWhVvTOm2mHydzLEs8,6157
-tests/test_db6.py,sha256=O19L1nZiQX492ts3uY3yU6dprzcH7WTUobJKb4HErFA,37411
-tests/test_mysetting.py,sha256=te6B_BErGc6ThgBpAl45n0Nd0hWJcm-X9WTw14bR6MA,5781
-tests/test_xml.py,sha256=SMRML85D24AMpgqBASK1rd-5UdacTlyUVWZGpOoRp40,16854
-pyrekordbox-0.3.1.dist-info/LICENSE,sha256=Au8sngdQ79q5JsZXZJVU13j46VDxFbPwC4x6sSz8Jes,1074
-pyrekordbox-0.3.1.dist-info/METADATA,sha256=5OaFdFRXPPnvMKhbifLndCDzWvi74uwq3offFgT25z0,16720
-pyrekordbox-0.3.1.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
-pyrekordbox-0.3.1.dist-info/top_level.txt,sha256=Cv8QDfcJ7y8fYUm0Q8D5GoiGxaqb7qt8Z5ntVbj1cLk,23
-pyrekordbox-0.3.1.dist-info/RECORD,,

tests/__init__.py

@@ -1,3 +0,0 @@
-# -*- coding: utf-8 -*-
-# Author: Dylan Jones
-# Date:   2022-05-07

tests/test_anlz.py

@@ -1,206 +0,0 @@
-# -*- coding: utf-8 -*-
-# Author: Dylan Jones
-# Date:   2023-02-01
-
-import os
-import pytest
-import numpy as np
-from numpy.testing import assert_equal
-from pyrekordbox import anlz
-
-TEST_ROOT = os.path.join(os.path.dirname(os.path.dirname(__file__)), ".testdata")
-ANLZ_ROOT = os.path.join(TEST_ROOT, "export", "PIONEER", "USBANLZ")
-ANLZ_DIRS = list(anlz.walk_anlz_paths(ANLZ_ROOT))
-ANLZ_FILES = [paths for _, paths in ANLZ_DIRS]
-
-
-def test_parse():
-    for root, files in ANLZ_DIRS:
-        for path in files.values():
-            anlz.AnlzFile.parse_file(path)
-
-
-def test_rebuild():
-    for root, files in ANLZ_DIRS:
-        for path in files.values():
-            file = anlz.AnlzFile.parse_file(path)
-            data = file.build()
-            assert len(data) == file.file_header.len_file
-            _ = anlz.AnlzFile.parse(data)
-
-
-def test_read_anlz_files():
-    for root, files in ANLZ_DIRS:
-        anlz_files = anlz.read_anlz_files(root)
-        assert len(files) == len(anlz_files)
-
-
-# -- Tags ------------------------------------------------------------------------------
-
-
-@pytest.mark.parametrize("paths", ANLZ_FILES)
-def test_pqtz_tag_getters(paths):
-    file = anlz.AnlzFile.parse_file(paths["DAT"])
-    tag = file.get_tag("PQTZ")
-    beats, bpms, times = tag.get()
-    nbeats = len(beats)
-    # Check that shape of arrays are equal
-    assert nbeats == len(bpms)
-    assert nbeats == len(times)
-    # Check that the beats array only contains the values 1-4
-    if nbeats:
-        assert_equal(np.sort(np.unique(beats)), [1, 2, 3, 4])
-
-    # Check other getters
-    assert_equal(beats, tag.get_beats())
-    assert_equal(bpms, tag.get_bpms())
-    assert_equal(times, tag.get_times())
-
-
-def test_pqtz_tag_set_beats():
-    paths = ANLZ_FILES[0]
-    file = anlz.AnlzFile.parse_file(paths["DAT"])
-    tag = file.get_tag("PQTZ")
-
-    beats = np.ones(tag.count)
-    tag.set_beats(beats)
-    assert_equal(tag.get_beats(), beats)
-
-
-def test_pqtz_tag_set_bpms():
-    paths = ANLZ_FILES[0]
-    file = anlz.AnlzFile.parse_file(paths["DAT"])
-    tag = file.get_tag("PQTZ")
-
-    bpms = 100 * np.ones(tag.count, dtype=np.float64)
-    tag.set_bpms(bpms)
-    assert_equal(tag.get_bpms(), bpms)
-
-
-def test_pqtz_tag_set_times():
-    paths = ANLZ_FILES[0]
-    file = anlz.AnlzFile.parse_file(paths["DAT"])
-    tag = file.get_tag("PQTZ")
-
-    times = 0.5 * np.arange(tag.count)
-    tag.set_times(times)
-    assert_equal(tag.get_times(), times)
-
-
-def test_pqtz_tag_set():
-    paths = ANLZ_FILES[0]
-    file = anlz.AnlzFile.parse_file(paths["DAT"])
-    tag = file.get_tag("PQTZ")
-
-    beats = np.ones(tag.count)
-    bpms = 100 * np.ones(tag.count, dtype=np.float64)
-    times = 0.5 * np.arange(tag.count)
-    tag.set(beats, bpms, times)
-    assert_equal(tag.get_bpms(), bpms)
-    assert_equal(tag.get_bpms(), bpms)
-    assert_equal(tag.get_times(), times)
-
-
-@pytest.mark.parametrize("paths", ANLZ_FILES)
-def test_ppth_tag_getters(paths):
-    tmp = ""
-    for path in paths.values():
-        file = anlz.AnlzFile.parse_file(path)
-        tag = file.get_tag("PPTH")
-        p = tag.get()
-        if not tmp:
-            tmp = p
-        else:
-            assert tmp == p
-
-
-def test_ppth_tag_setters():
-    paths = ANLZ_FILES[0]
-    extected = r"C:/new/path/to/file.mp3"
-    for path in paths.values():
-        file = anlz.AnlzFile.parse_file(path)
-        tag = file.get_tag("PPTH")
-        tag.set(r"C:\new\path\to\file.mp3")
-        assert tag.get() == extected
-
-
-@pytest.mark.parametrize("paths", ANLZ_FILES)
-def test_pwav_tag_getters(paths):
-    file = anlz.AnlzFile.parse_file(paths["DAT"])
-    tag = file.get_tag("PWAV")
-
-    heights, color = tag.get()
-    assert len(heights) == len(color)
-    assert np.all(np.logical_and(0 <= heights, heights <= 31))
-    assert np.all(np.logical_and(0 <= color, color <= 7))
-
-
-@pytest.mark.parametrize("paths", ANLZ_FILES)
-def test_pwv2_tag_getters(paths):
-    file = anlz.AnlzFile.parse_file(paths["DAT"])
-    tag = file.get_tag("PWV2")
-
-    heights, color = tag.get()
-    assert len(heights) == len(color)
-    assert np.all(np.logical_and(0 <= heights, heights <= 31))
-    assert np.all(np.logical_and(0 <= color, color <= 7))
-
-
-@pytest.mark.parametrize("paths", ANLZ_FILES)
-def test_pwv3_tag_getters(paths):
-    file = anlz.AnlzFile.parse_file(paths["EXT"])
-    tag = file.get_tag("PWV3")
-
-    heights, color = tag.get()
-    assert len(heights) == len(color)
-    assert np.all(np.logical_and(0 <= heights, heights <= 31))
-    assert np.all(np.logical_and(0 <= color, color <= 7))
-
-
-@pytest.mark.parametrize("paths", ANLZ_FILES)
-def test_pwv4_tag_getters(paths):
-    file = anlz.AnlzFile.parse_file(paths["EXT"])
-    tag = file.get_tag("PWV4")
-
-    heights, colors, blues = tag.get()
-    assert len(heights) == len(colors)
-    assert len(heights) == len(blues)
-
-
-@pytest.mark.parametrize("paths", ANLZ_FILES)
-def test_pwv5_tag_getters(paths):
-    file = anlz.AnlzFile.parse_file(paths["EXT"])
-    tag = file.get_tag("PWV5")
-
-    heights, colors = tag.get()
-    assert len(heights) == colors.shape[0]
-
-
-# -- File ------------------------------------------------------------------------------
-
-
-def test_anlzfile_getall_tags():
-    paths = ANLZ_FILES[0]
-    file = anlz.AnlzFile.parse_file(paths["DAT"])
-    key = "PPTH"
-    tags = file.getall_tags(key)
-    assert len(tags) == 1
-    assert tags[0].get() == file.get(key)
-
-
-def test_anlzfile_get():
-    paths = ANLZ_FILES[0]
-    file = anlz.AnlzFile.parse_file(paths["DAT"])
-    key = "PPTH"
-    tag = file.get_tag(key)
-    assert file.get(key) == tag.get()
-
-
-def test_anlzfile_getall():
-    paths = ANLZ_FILES[0]
-    file = anlz.AnlzFile.parse_file(paths["DAT"])
-    key = "PPTH"
-    tag = file.get_tag(key)
-    values = file.getall(key)
-    assert len(values) == 1
-    assert values[0] == tag.get()