pyrekordbox 0.3.2__py3-none-any.whl → 0.4.1__py3-none-any.whl

pyrekordbox/db6/tables.py

@@ -5,15 +5,28 @@
 """Rekordbox 6 `master.db` SQLAlchemy table declarations."""
 
 import math
+import re
 import struct
-import numpy as np
-from enum import IntEnum
 from datetime import datetime
-from sqlalchemy import Column, Integer, VARCHAR, BigInteger, SmallInteger, Text, Float
-from sqlalchemy import ForeignKey, TypeDecorator
-from sqlalchemy.orm import DeclarativeBase, relationship, backref, mapped_column, Mapped
+from enum import IntEnum
+from typing import List
+
+import numpy as np
+from sqlalchemy import (
+    VARCHAR,
+    BigInteger,
+    Column,
+    Float,
+    ForeignKey,
+    Integer,
+    SmallInteger,
+    Text,
+    TypeDecorator,
+)
 from sqlalchemy.ext.associationproxy import association_proxy
 from sqlalchemy.inspection import inspect
+from sqlalchemy.orm import DeclarativeBase, Mapped, backref, mapped_column, relationship
+
 from .registry import RekordboxAgentRegistry
 
 __all__ = [
@@ -102,34 +115,53 @@ TABLES = [
 ]
 
 
+def datetime_to_str(value: datetime) -> str:
+    s = value.isoformat().replace("T", " ")
+    if value.tzinfo is not None:
+        # Get the timezone info (last 6 characters of the string)
+        tzinfo = s[-6:]
+        s = s[:-9] + " " + tzinfo
+    else:
+        s = s[:-3] + " +00:00"
+    return s
+
+
+def string_to_datetime(value: str) -> datetime:
+    try:
+        dt = datetime.fromisoformat(value)
+    except ValueError:
+        if len(value.strip()) > 23:
+            # Assume the format
+            # "2025-04-12 19:11:29.274 -05:00" or
+            # "2025-04-12 19:11:29.274 -05:00 (Central Daylight Time)"
+            datestr, tzinfo = value[:23], value[23:30]
+            datestr = datestr.strip()
+            tzinfo = tzinfo.strip()
+            assert re.match(r"^[+-]?\d{1,2}:\d{2}", tzinfo)
+            datestr = datestr.strip() + tzinfo
+        else:
+            datestr, tzinfo = value, ""
+        dt = datetime.fromisoformat(datestr)
+    # Convert to local timezone and return without timezone
+    return dt.astimezone().replace(tzinfo=None)
+
+
 class DateTime(TypeDecorator):
     """Custom datetime column with timezone support.
 
     The datetime format in the database is `YYYY-MM-DD HH:MM:SS.SSS +00:00`.
-    The timezone seems to always be `+00:00` (UTC).
     This format is not supported by the `DateTime` column of SQLAlchemy 2.
     """
 
     impl = Text
     cache_ok = True
 
-    def process_bind_param(self, value, dialect):
-        return value.isoformat().replace("T", " ")[:-3] + " +00:00"
+    def process_bind_param(self, value: datetime, dialect) -> str:
+        return datetime_to_str(value)
 
-    def process_result_value(self, value, dialect):
+    def process_result_value(self, value: str, dialect):
         if value:
-            try:
-                dt = datetime.fromisoformat(value)
-            except ValueError:
-                if len(value.strip()) > 23:
-                    datestr, tzinfo = value[:23], value[23:]
-                    datestr = datestr.strip()
-                    tzinfo = tzinfo.strip()
-                    assert tzinfo == "+00:00", tzinfo
-                else:
-                    datestr, tzinfo = value, ""
-                dt = datetime.fromisoformat(datestr)
-            return dt
+            return string_to_datetime(value)
         return None
 
 
@@ -139,6 +171,15 @@ class PlaylistType(IntEnum):
     SMART_PLAYLIST = 4
 
 
+class FileType(IntEnum):
+    MP3 = 1
+    M4A = 4
+    FLAC = 5
+    WAV = 11
+    AIFF = 12
+    AIF = 12
+
+
 # -- Base- and Mixin classes -----------------------------------------------------------
 
 
@@ -146,6 +187,7 @@ class Base(DeclarativeBase):
     """Base class used to initialize the declarative base for all tables."""
 
     __tablename__: str
+    __keys__: List[str] = []
 
     @classmethod
     def create(cls, **kwargs):
@@ -164,13 +206,23 @@ class Base(DeclarativeBase):
         """Returns a list of all relationship names."""
         return [column.key for column in inspect(cls).relationships]  # noqa
 
+    @classmethod
+    def __get_keys__(cls):
+        """Get all attributes of the table."""
+        items = cls.__dict__.items()
+        keys = [k for k, v in items if not callable(v) and not k.startswith("_")]
+        return keys
+
+    @classmethod
+    def keys(cls):
+        """Returns a list of all column names including the relationships."""
+        if not cls.__keys__:  # Cache the keys
+            cls.__keys__ = cls.__get_keys__()
+        return cls.__keys__
+
     def __iter__(self):
         """Iterates over all columns and relationship names."""
-        insp = inspect(self.__class__)
-        for column in insp.c:
-            yield column.name
-        for column in insp.relationships:  # noqa
-            yield column.key
+        return iter(self.keys())
 
     def __len__(self):
         return sum(1 for _ in self.__iter__())
@@ -184,10 +236,6 @@ class Base(DeclarativeBase):
             RekordboxAgentRegistry.on_update(self, key, value)
         super().__setattr__(key, value)
 
-    def keys(self):
-        """Returns a list of all column names including the relationships."""
-        return list(self.__iter__())
-
     def values(self):
         """Returns a list of all column values including the relationships."""
         return [self.__getitem__(key) for key in self.keys()]
@@ -212,9 +260,7 @@ class Base(DeclarativeBase):
 class StatsTime:
     """Mixin class for tables that only use time statistics columns."""
 
-    created_at: Mapped[datetime] = mapped_column(
-        DateTime, nullable=False, default=datetime.now
-    )
+    created_at: Mapped[datetime] = mapped_column(DateTime, nullable=False, default=datetime.now)
     """The creation date of the table entry (from :class:`StatsTime`)."""
     updated_at: Mapped[datetime] = mapped_column(
         DateTime, nullable=False, default=datetime.now, onupdate=datetime.now
@@ -243,9 +289,7 @@ class StatsFull:
     rb_local_usn: Mapped[int] = mapped_column(BigInteger, default=None)
     """The local USN (unique sequence number) of the table entry
     (from :class:`StatsFull`)."""
-    created_at: Mapped[datetime] = mapped_column(
-        DateTime, nullable=False, default=datetime.now
-    )
+    created_at: Mapped[datetime] = mapped_column(DateTime, nullable=False, default=datetime.now)
     """The creation date of the table entry (from :class:`StatsFull`)."""
     updated_at: Mapped[datetime] = mapped_column(
         DateTime, nullable=False, default=datetime.now, onupdate=datetime.now
@@ -343,9 +387,7 @@ class ContentActiveCensor(Base, StatsFull):
 
     ID: Mapped[str] = mapped_column(VARCHAR(255), primary_key=True)
     """The ID (primary key) of the table entry."""
-    ContentID: Mapped[str] = mapped_column(
-        VARCHAR(255), ForeignKey("djmdContent.ID"), default=None
-    )
+    ContentID: Mapped[str] = mapped_column(VARCHAR(255), ForeignKey("djmdContent.ID"), default=None)
     """The ID of the :class:`DjmdContent` entry this censor belongs to."""
     ActiveCensors: Mapped[str] = mapped_column(Text, default=None)
     """The active censors of the table entry."""
@@ -368,9 +410,7 @@ class ContentCue(Base, StatsFull):
 
     ID: Mapped[str] = mapped_column(VARCHAR(255), primary_key=True)
     """The ID (primary key) of the table entry."""
-    ContentID: Mapped[str] = mapped_column(
-        VARCHAR(255), ForeignKey("djmdContent.ID"), default=None
-    )
+    ContentID: Mapped[str] = mapped_column(VARCHAR(255), ForeignKey("djmdContent.ID"), default=None)
     """The ID of the :class:`DjmdContent` entry this cue belongs to."""
     Cues: Mapped[str] = mapped_column(Text, default=None)
     """The cues of the table entry."""
@@ -393,9 +433,7 @@ class ContentFile(Base, StatsFull):
 
     ID: Mapped[str] = mapped_column(VARCHAR(255), primary_key=True)
     """The ID (primary key) of the table entry."""
-    ContentID: Mapped[str] = mapped_column(
-        VARCHAR(255), ForeignKey("djmdContent.ID"), default=None
-    )
+    ContentID: Mapped[str] = mapped_column(VARCHAR(255), ForeignKey("djmdContent.ID"), default=None)
     """The ID of the :class:`DjmdContent` entry this file belongs to."""
     Path: Mapped[str] = mapped_column(VARCHAR(255), default=None)
     """The path of the file."""
@@ -440,9 +478,7 @@ class DjmdActiveCensor(Base, StatsFull):
 
     ID: Mapped[str] = mapped_column(VARCHAR(255), primary_key=True)
     """The ID (primary key) of the table entry."""
-    ContentID: Mapped[str] = mapped_column(
-        VARCHAR(255), ForeignKey("djmdContent.ID"), default=None
-    )
+    ContentID: Mapped[str] = mapped_column(VARCHAR(255), ForeignKey("djmdContent.ID"), default=None)
     """The ID of the :class:`DjmdContent` entry this censor belongs to."""
     InMsec: Mapped[int] = mapped_column(Integer, default=None)
     """The in time of the censor (in milliseconds)."""
@@ -455,9 +491,7 @@ class DjmdActiveCensor(Base, StatsFull):
     ContentUUID: Mapped[str] = mapped_column(VARCHAR(255), default=None)
     """The UUID of the :class:`DjmdContent` entry this censor belongs to."""
 
-    Content = relationship(
-        "DjmdContent", foreign_keys=ContentID, back_populates="ActiveCensors"
-    )
+    Content = relationship("DjmdContent", foreign_keys=ContentID, back_populates="ActiveCensors")
     """The content entry this censor belongs to (links to :class:`DjmdContent`)."""
 
 
@@ -588,17 +622,11 @@ class DjmdContent(Base, StatsFull):
     """The short file name of the file corresponding to the content entry."""
     Title: Mapped[str] = mapped_column(VARCHAR(255), default=None)
     """The title of the track."""
-    ArtistID: Mapped[str] = mapped_column(
-        VARCHAR(255), ForeignKey("djmdArtist.ID"), default=None
-    )
+    ArtistID: Mapped[str] = mapped_column(VARCHAR(255), ForeignKey("djmdArtist.ID"), default=None)
     """The ID of the :class:`DjmdArtist` entry of the artist of this track."""
-    AlbumID: Mapped[str] = mapped_column(
-        VARCHAR(255), ForeignKey("djmdAlbum.ID"), default=None
-    )
+    AlbumID: Mapped[str] = mapped_column(VARCHAR(255), ForeignKey("djmdAlbum.ID"), default=None)
     """The ID of the :class:`DjmdAlbum` entry of the album of this track."""
-    GenreID: Mapped[str] = mapped_column(
-        VARCHAR(255), ForeignKey("djmdGenre.ID"), default=None
-    )
+    GenreID: Mapped[str] = mapped_column(VARCHAR(255), ForeignKey("djmdGenre.ID"), default=None)
     """The ID of the :class:`DjmdGenre` entry of the genre of this track."""
     BPM: Mapped[int] = mapped_column(Integer, default=None)
     """The BPM (beats per minute) of the track."""
@@ -618,27 +646,19 @@ class DjmdContent(Base, StatsFull):
     """The rating of the track."""
     ReleaseYear: Mapped[int] = mapped_column(Integer, default=None)
     """The release year of the track."""
-    RemixerID: Mapped[str] = mapped_column(
-        VARCHAR(255), ForeignKey("djmdArtist.ID"), default=None
-    )
+    RemixerID: Mapped[str] = mapped_column(VARCHAR(255), ForeignKey("djmdArtist.ID"), default=None)
     """The ID of the :class:`DjmdArtist` entry of the remixer of this track."""
-    LabelID: Mapped[str] = mapped_column(
-        VARCHAR(255), ForeignKey("djmdLabel.ID"), default=None
-    )
+    LabelID: Mapped[str] = mapped_column(VARCHAR(255), ForeignKey("djmdLabel.ID"), default=None)
     """The ID of the :class:`DjmdLabel` entry of the label of this track."""
     OrgArtistID: Mapped[str] = mapped_column(
         VARCHAR(255), ForeignKey("djmdArtist.ID"), default=None
     )
     """The ID of the :class:`DjmdArtist` entry of the original artist of this track."""
-    KeyID: Mapped[str] = mapped_column(
-        VARCHAR(255), ForeignKey("djmdKey.ID"), default=None
-    )
+    KeyID: Mapped[str] = mapped_column(VARCHAR(255), ForeignKey("djmdKey.ID"), default=None)
     """The ID of the :class:`DjmdKey` entry of the key of this track."""
     StockDate: Mapped[str] = mapped_column(VARCHAR(255), default=None)
     """The stock date of the track."""
-    ColorID: Mapped[str] = mapped_column(
-        VARCHAR(255), ForeignKey("djmdColor.ID"), default=None
-    )
+    ColorID: Mapped[str] = mapped_column(VARCHAR(255), ForeignKey("djmdColor.ID"), default=None)
     """The ID of the :class:`DjmdColor` entry of the color of this track."""
     DJPlayCount: Mapped[str] = mapped_column(VARCHAR(255), default=None)
     """The play count of the track."""
@@ -656,9 +676,7 @@ class DjmdContent(Base, StatsFull):
     """The file size of the track."""
     DiscNo: Mapped[int] = mapped_column(Integer, default=None)
     """The number of the disc of the album of the track."""
-    ComposerID: Mapped[str] = mapped_column(
-        VARCHAR(255), ForeignKey("djmdArtist.ID"), default=None
-    )
+    ComposerID: Mapped[str] = mapped_column(VARCHAR(255), ForeignKey("djmdArtist.ID"), default=None)
     """The ID of the :class:`DjmdArtist` entry of the composer of this track."""
     Subtitle: Mapped[str] = mapped_column(VARCHAR(255), default=None)
     """The subtitle of the track."""
@@ -690,9 +708,7 @@ class DjmdContent(Base, StatsFull):
     """The analysis updated status of the track."""
     TrackInfoUpdated: Mapped[str] = mapped_column(VARCHAR(255), default=None)
     """The track info updated status of the track."""
-    Lyricist: Mapped[str] = mapped_column(
-        VARCHAR(255), ForeignKey("djmdArtist.ID"), default=None
-    )
+    Lyricist: Mapped[str] = mapped_column(VARCHAR(255), ForeignKey("djmdArtist.ID"), default=None)
     """The ID of the :class:`DjmdArtist` entry of the lyricist of this track."""
     ISRC: Mapped[str] = mapped_column(VARCHAR(255), default=None)
     """The ISRC of the track."""
@@ -759,9 +775,7 @@ class DjmdContent(Base, StatsFull):
     """The album artist entry of the track (links to :class:`DjmdArtist`)."""
     MyTags = relationship("DjmdSongMyTag", back_populates="Content")
     """The my tags of the track (links to :class:`DjmdSongMyTag`)."""
-    Cues = relationship(
-        "DjmdCue", foreign_keys="DjmdCue.ContentID", back_populates="Content"
-    )
+    Cues = relationship("DjmdCue", foreign_keys="DjmdCue.ContentID", back_populates="Content")
     """The cues of the track (links to :class:`DjmdCue`)."""
     ActiveCensors = relationship("DjmdActiveCensor", back_populates="Content")
     """The active censors of the track (links to :class:`DjmdActiveCensor`)."""
@@ -810,9 +824,7 @@ class DjmdCue(Base, StatsFull):
 
     ID: Mapped[str] = mapped_column(VARCHAR(255), primary_key=True)
     """The ID (primary key) of the table entry."""
-    ContentID: Mapped[str] = mapped_column(
-        VARCHAR(255), ForeignKey("djmdContent.ID"), default=None
-    )
+    ContentID: Mapped[str] = mapped_column(VARCHAR(255), ForeignKey("djmdContent.ID"), default=None)
     """The ID of the content (:class:`DjmdContent`) containing the cue point."""
     InMsec: Mapped[int] = mapped_column(Integer, default=None)
     """The in point of the cue point in milliseconds."""
@@ -915,9 +927,7 @@ class DjmdHistory(Base, StatsFull):
     """The name of the history playlist."""
     Attribute: Mapped[int] = mapped_column(Integer, default=None)
     """The attributes of the history playlist"""
-    ParentID: Mapped[str] = mapped_column(
-        VARCHAR(255), ForeignKey("djmdHistory.ID"), default=None
-    )
+    ParentID: Mapped[str] = mapped_column(VARCHAR(255), ForeignKey("djmdHistory.ID"), default=None)
     """The ID of the parent history playlist (:class:`DjmdHistory`)."""
     DateCreated: Mapped[str] = mapped_column(VARCHAR(255), default=None)
     """The date the history playlist was created."""
@@ -950,13 +960,9 @@ class DjmdSongHistory(Base, StatsFull):
 
     ID: Mapped[str] = mapped_column(VARCHAR(255), primary_key=True)
 
-    HistoryID: Mapped[str] = mapped_column(
-        VARCHAR(255), ForeignKey("djmdHistory.ID"), default=None
-    )
+    HistoryID: Mapped[str] = mapped_column(VARCHAR(255), ForeignKey("djmdHistory.ID"), default=None)
     """The ID of the history playlist (:class:`DjmdHistory`)."""
-    ContentID: Mapped[str] = mapped_column(
-        VARCHAR(255), ForeignKey("djmdContent.ID"), default=None
-    )
+    ContentID: Mapped[str] = mapped_column(VARCHAR(255), ForeignKey("djmdContent.ID"), default=None)
     """The ID of the content (:class:`DjmdContent`)."""
     TrackNo: Mapped[int] = mapped_column(Integer, default=None)
     """The track number of the song in the history playlist."""
@@ -1022,9 +1028,7 @@ class DjmdSongHotCueBanklist(Base, StatsFull):
         VARCHAR(255), ForeignKey("djmdHotCueBanklist.ID"), default=None
     )
     """The ID of the hot cue banklist (:class:`DjmdHotCueBanklist`)."""
-    ContentID: Mapped[str] = mapped_column(
-        VARCHAR(255), ForeignKey("djmdContent.ID"), default=None
-    )
+    ContentID: Mapped[str] = mapped_column(VARCHAR(255), ForeignKey("djmdContent.ID"), default=None)
     """The ID of the content (:class:`DjmdContent`)."""
     TrackNo: Mapped[int] = mapped_column(Integer, default=None)
     """The track number of the hot-cue in the hot cue banklist."""
@@ -1132,9 +1136,7 @@ class DjmdMixerParam(Base, StatsFull):
 
     ID: Mapped[str] = mapped_column(VARCHAR(255), primary_key=True)
     """The ID (primary key) of the table entry."""
-    ContentID: Mapped[str] = mapped_column(
-        VARCHAR(255), ForeignKey("djmdContent.ID"), default=None
-    )
+    ContentID: Mapped[str] = mapped_column(VARCHAR(255), ForeignKey("djmdContent.ID"), default=None)
     """The ID of the content (:class:`DjmdContent`)."""
     GainHigh: Mapped[int] = mapped_column(Integer, default=None)
     """The high gain of the mixer parameter."""
@@ -1215,9 +1217,7 @@ class DjmdMyTag(Base, StatsFull):
     """The name of the My-Tag list."""
     Attribute: Mapped[int] = mapped_column(Integer, default=None)
     """The attribute of the My-Tag list."""
-    ParentID: Mapped[str] = mapped_column(
-        VARCHAR(255), ForeignKey("djmdMyTag.ID"), default=None
-    )
+    ParentID: Mapped[str] = mapped_column(VARCHAR(255), ForeignKey("djmdMyTag.ID"), default=None)
     """The ID of the parent My-Tag list (:class:`DjmdMyTag`)."""
 
     MyTags = relationship("DjmdSongMyTag", back_populates="MyTag")
@@ -1246,13 +1246,9 @@ class DjmdSongMyTag(Base, StatsFull):
 
     ID: Mapped[str] = mapped_column(VARCHAR(255), primary_key=True)
     """The ID (primary key) of the table entry."""
-    MyTagID: Mapped[str] = mapped_column(
-        VARCHAR(255), ForeignKey("djmdMyTag.ID"), default=None
-    )
+    MyTagID: Mapped[str] = mapped_column(VARCHAR(255), ForeignKey("djmdMyTag.ID"), default=None)
     """The ID of the My-Tag list (links to :class:`DjmdMyTag`)."""
-    ContentID: Mapped[str] = mapped_column(
-        VARCHAR(255), ForeignKey("djmdContent.ID"), default=None
-    )
+    ContentID: Mapped[str] = mapped_column(VARCHAR(255), ForeignKey("djmdContent.ID"), default=None)
     """The ID of the content this item belongs to (:class:`DjmdContent`)."""
     TrackNo: Mapped[int] = mapped_column(Integer, default=None)
     """The track number of the My-Tag item (for ordering)."""
@@ -1286,16 +1282,12 @@ class DjmdPlaylist(Base, StatsFull):
     """The path to the image of the playlist."""
     Attribute: Mapped[int] = mapped_column(Integer, default=None)
     """The attribute of the playlist."""
-    ParentID: Mapped[str] = mapped_column(
-        VARCHAR(255), ForeignKey("djmdPlaylist.ID"), default=None
-    )
+    ParentID: Mapped[str] = mapped_column(VARCHAR(255), ForeignKey("djmdPlaylist.ID"), default=None)
     """The ID of the parent playlist (:class:`DjmdPlaylist`)."""
     SmartList: Mapped[str] = mapped_column(Text, default=None)
     """The smart list settings of the playlist."""
 
-    Songs = relationship(
-        "DjmdSongPlaylist", back_populates="Playlist", cascade="all, delete"
-    )
+    Songs = relationship("DjmdSongPlaylist", back_populates="Playlist", cascade="all, delete")
     """The contents of the playlist (links to :class:`DjmdSongPlaylist`)."""
     Children = relationship(
         "DjmdPlaylist",
@@ -1340,9 +1332,7 @@ class DjmdSongPlaylist(Base, StatsFull):
         VARCHAR(255), ForeignKey("djmdPlaylist.ID"), default=None
     )
     """The ID of the playlist this item is in (:class:`DjmdPlaylist`)."""
-    ContentID: Mapped[str] = mapped_column(
-        VARCHAR(255), ForeignKey("djmdContent.ID"), default=None
-    )
+    ContentID: Mapped[str] = mapped_column(VARCHAR(255), ForeignKey("djmdContent.ID"), default=None)
     """The ID of the content this item belongs to (:class:`DjmdContent`)."""
     TrackNo: Mapped[int] = mapped_column(Integer, default=None)
     """The track number of the playlist item (for ordering)."""
@@ -1413,9 +1403,7 @@ class DjmdSongRelatedTracks(Base, StatsFull):
     )
     """The ID of the related tracks list this item is in
     (:class:`DjmdRelatedTracks`)."""
-    ContentID: Mapped[str] = mapped_column(
-        VARCHAR(255), ForeignKey("djmdContent.ID"), default=None
-    )
+    ContentID: Mapped[str] = mapped_column(VARCHAR(255), ForeignKey("djmdContent.ID"), default=None)
     """The ID of the content this item belongs to (:class:`DjmdContent`)."""
     TrackNo: Mapped[int] = mapped_column(Integer, default=None)
     """The track number of the related tracks list item (for ordering)."""
@@ -1444,9 +1432,7 @@ class DjmdSampler(Base, StatsFull):
     """The name of the sampler list."""
     Attribute: Mapped[int] = mapped_column(Integer, default=None)
     """The attribute of the sampler list."""
-    ParentID: Mapped[str] = mapped_column(
-        VARCHAR(255), ForeignKey("djmdSampler.ID"), default=None
-    )
+    ParentID: Mapped[str] = mapped_column(VARCHAR(255), ForeignKey("djmdSampler.ID"), default=None)
     """The ID of the parent sampler list (:class:`DjmdSampler`)."""
 
     Songs = relationship("DjmdSongSampler", back_populates="Sampler")
@@ -1477,13 +1463,9 @@ class DjmdSongSampler(Base, StatsFull):
 
     ID: Mapped[str] = mapped_column(VARCHAR(255), primary_key=True)
     """The ID (primary key) of the table entry."""
-    SamplerID: Mapped[str] = mapped_column(
-        VARCHAR(255), ForeignKey("djmdSampler.ID"), default=None
-    )
+    SamplerID: Mapped[str] = mapped_column(VARCHAR(255), ForeignKey("djmdSampler.ID"), default=None)
     """The ID of the sampler list this item is in (:class:`DjmdSampler`)."""
-    ContentID: Mapped[str] = mapped_column(
-        VARCHAR(255), ForeignKey("djmdContent.ID"), default=None
-    )
+    ContentID: Mapped[str] = mapped_column(VARCHAR(255), ForeignKey("djmdContent.ID"), default=None)
     """The ID of the content this item belongs to (:class:`DjmdContent`)."""
     TrackNo: Mapped[int] = mapped_column(Integer, default=None)
     """The track number of the sampler list item (for ordering)."""
@@ -1501,9 +1483,7 @@ class DjmdSongTagList(Base, StatsFull):
 
     ID: Mapped[str] = mapped_column(VARCHAR(255), primary_key=True)
     """The ID (primary key) of the table entry."""
-    ContentID: Mapped[str] = mapped_column(
-        VARCHAR(255), ForeignKey("djmdContent.ID"), default=None
-    )
+    ContentID: Mapped[str] = mapped_column(VARCHAR(255), ForeignKey("djmdContent.ID"), default=None)
     """The ID of the content this item belongs to (:class:`DjmdContent`)."""
     TrackNo: Mapped[int] = mapped_column(Integer, default=None)
     """The track number of the tag list item (for ordering)."""

pyrekordbox/logger.py

@@ -20,4 +20,3 @@ logger.addHandler(sh)
 
 # Set logging level
 logger.setLevel(logging.WARNING)
-logging.root.setLevel(logging.NOTSET)

pyrekordbox/mysettings/__init__.py

@@ -4,14 +4,15 @@
 
 import re
 from pathlib import Path
+
 from . import structs
 from .file import (
     FILES,
-    SettingsFile,
-    MySettingFile,
-    MySetting2File,
-    DjmMySettingFile,
     DevSettingFile,
+    DjmMySettingFile,
+    MySetting2File,
+    MySettingFile,
+    SettingsFile,
 )
 
 RE_MYSETTING = re.compile(".*SETTING[0-9]?.DAT$")

pyrekordbox/mysettings/file.py

@@ -5,8 +5,10 @@
 """Rekordbox My-Setting file handlers."""
 
 import re
-from construct import Struct
 from collections.abc import MutableMapping
+
+from construct import Struct
+
 from . import structs
 
 # fmt: off

pyrekordbox/rbxml.py

@@ -7,10 +7,12 @@ r"""Rekordbox XML database file handler."""
 import logging
 import os.path
 import urllib.parse
+import xml.etree.cElementTree as xml
 from abc import abstractmethod
 from collections import abc
-import xml.etree.cElementTree as xml
+
 import bidict
+
 from .utils import pretty_xml
 
 logger = logging.getLogger(__name__)
@@ -25,9 +27,7 @@ POSMARK_TYPE_MAPPING = bidict.bidict(
         "4": "loop",
     }
 )
-RATING_MAPPING = bidict.bidict(
-    {"0": 0, "51": 1, "102": 2, "153": 3, "204": 4, "255": 5}
-)
+RATING_MAPPING = bidict.bidict({"0": 0, "51": 1, "102": 2, "153": 3, "204": 4, "255": 5})
 NODE_KEYTYPE_MAPPING = bidict.bidict({"0": "TrackID", "1": "Location"})
 
 
@@ -271,9 +271,7 @@ class Tempo(AbstractElement):
     ATTRIBS = ["Inizio", "Bpm", "Metro", "Battito"]
     GETTERS = {"Inizio": float, "Bpm": float, "Battito": int}
 
-    def __init__(
-        self, parent=None, Inizio=0.0, Bpm=0.0, Metro="4/4", Battito=1, element=None
-    ):
+    def __init__(self, parent=None, Inizio=0.0, Bpm=0.0, Metro="4/4", Battito=1, element=None):
         super().__init__(element, parent, Inizio, Bpm, Metro, Battito)
 
     def _init(self, parent, inizio, bpm, metro, battito):
@@ -658,7 +656,7 @@ class Node:
 
     @property
     def key_type(self):
-        """str: The type of key used by the playlist node"""
+        """str: The type of key used by the playlist node."""
         return NODE_KEYTYPE_MAPPING.get(self._element.attrib.get("KeyType"))
 
     @property
@@ -1262,9 +1260,7 @@ class RekordboxXml:
         num_tracks = len(self._collection.findall(f".//{Track.TAG}"))
         n = int(self._collection.attrib["Entries"])
         if n != num_tracks:
-            raise ValueError(
-                f"Track count {num_tracks} does not match number of elements {n}"
-            )
+            raise ValueError(f"Track count {num_tracks} does not match number of elements {n}")
         # Generate XML string
         return pretty_xml(self._root, indent, encoding="utf-8")
 
@@ -1280,7 +1276,7 @@ class RekordboxXml:
             The default is 3 spaces.
         """
         string = self.tostring(indent)
-        with open(path, "w") as fh:
+        with open(path, "w", encoding="utf-8") as fh:
             fh.write(string)
 
     def __repr__(self):

pyrekordbox/utils.py

@@ -6,9 +6,10 @@
 
 import os
 import warnings
-import psutil
-from xml.dom import minidom
 import xml.etree.cElementTree as xml
+from xml.dom import minidom
+
+import psutil
 
 warnings.simplefilter("always", DeprecationWarning)
 
@@ -122,7 +123,7 @@ def get_rekordbox_agent_pid(raise_exec=False):
 
 
 def pretty_xml(element, indent=None, encoding="utf-8"):
-    """Generates a formatted string of an XML element.
+    r"""Generates a formatted string of an XML element.
 
     Parameters
     ----------