2022-12-06 13:45:18 +00:00
|
|
|
import os
|
|
|
|
from typing import List
|
|
|
|
from mutagen.easyid3 import EasyID3
|
|
|
|
|
|
|
|
from ...utils.shared import (
|
|
|
|
MUSIC_DIR,
|
|
|
|
DATABASE_LOGGER as logger
|
|
|
|
)
|
|
|
|
from .database_object import (
|
|
|
|
DatabaseObject,
|
|
|
|
Reference
|
|
|
|
)
|
|
|
|
|
|
|
|
|
2022-12-06 22:44:42 +00:00
|
|
|
class SongAttribute:
|
|
|
|
def __init__(self, song_ref: Reference = None):
|
|
|
|
# the reference to the song the lyrics belong to
|
|
|
|
self.song_ref = song_ref
|
|
|
|
|
|
|
|
def add_song(self, song_ref: Reference):
|
|
|
|
self.song_ref = song_ref
|
|
|
|
|
|
|
|
def get_ref_song_id(self):
|
|
|
|
if self.song_ref is None:
|
|
|
|
return None
|
|
|
|
return self.song_ref.id
|
|
|
|
|
|
|
|
def set_ref_song_id(self, song_id):
|
|
|
|
self.song_ref = Reference(song_id)
|
|
|
|
|
|
|
|
song_ref_id = property(fget=get_ref_song_id, fset=set_ref_song_id)
|
|
|
|
|
|
|
|
|
2022-12-06 13:45:18 +00:00
|
|
|
class Metadata:
|
|
|
|
"""
|
|
|
|
Shall only be read or edited via the Song object.
|
|
|
|
For this reason there is no reference to the song needed.
|
|
|
|
"""
|
2022-12-06 22:44:42 +00:00
|
|
|
|
2022-12-06 13:45:18 +00:00
|
|
|
def __init__(self, data: dict = {}) -> None:
|
2022-12-06 22:44:42 +00:00
|
|
|
self.data = data
|
2022-12-06 13:45:18 +00:00
|
|
|
|
|
|
|
def get_all_metadata(self):
|
|
|
|
return list(self.data.items())
|
|
|
|
|
|
|
|
def __setitem__(self, item, value):
|
|
|
|
if item in EasyID3.valid_keys.keys():
|
|
|
|
self.data[item] = value
|
|
|
|
|
|
|
|
def __getitem__(self, item):
|
|
|
|
if item not in self.data:
|
|
|
|
return None
|
|
|
|
return self.data[item]
|
|
|
|
|
|
|
|
|
2022-12-06 22:44:42 +00:00
|
|
|
class Source(DatabaseObject, SongAttribute):
|
2022-12-06 13:45:18 +00:00
|
|
|
"""
|
|
|
|
create somehow like that
|
|
|
|
```python
|
|
|
|
# url won't be a valid one due to it being just an example
|
|
|
|
Source(src="youtube", url="https://youtu.be/dfnsdajlhkjhsd")
|
|
|
|
```
|
|
|
|
"""
|
2022-12-06 22:44:42 +00:00
|
|
|
|
2022-12-06 13:45:18 +00:00
|
|
|
def __init__(self, id_: str = None, src: str = None, url: str = None) -> None:
|
2022-12-06 22:44:42 +00:00
|
|
|
DatabaseObject.__init__(self, id_=id_)
|
|
|
|
SongAttribute.__init__(self)
|
2022-12-06 13:45:18 +00:00
|
|
|
|
|
|
|
self.src = src
|
|
|
|
self.url = url
|
|
|
|
|
2022-12-06 22:44:42 +00:00
|
|
|
def __str__(self):
|
|
|
|
return f"{self.src}: {self.url}"
|
|
|
|
|
2022-12-06 13:45:18 +00:00
|
|
|
|
2022-12-06 22:44:42 +00:00
|
|
|
class Target(DatabaseObject, SongAttribute):
|
2022-12-06 13:45:18 +00:00
|
|
|
"""
|
|
|
|
create somehow like that
|
|
|
|
```python
|
2022-12-06 22:44:42 +00:00
|
|
|
# I know path is pointless, and I will change that (don't worry about backwards compatibility there)
|
2022-12-06 13:45:18 +00:00
|
|
|
Target(file="~/Music/genre/artist/album/song.mp3", path="~/Music/genre/artist/album")
|
|
|
|
```
|
|
|
|
"""
|
2022-12-06 22:44:42 +00:00
|
|
|
|
|
|
|
def __init__(self, id_: str = None, file: str = None, path: str = None) -> None:
|
|
|
|
DatabaseObject.__init__(self, id_=id_)
|
|
|
|
SongAttribute.__init__(self)
|
2022-12-06 13:45:18 +00:00
|
|
|
self._file = file
|
|
|
|
self._path = path
|
|
|
|
|
|
|
|
def set_file(self, _file: str):
|
|
|
|
self._file = _file
|
|
|
|
|
|
|
|
def get_file(self) -> str | None:
|
|
|
|
if self._file is None:
|
|
|
|
return None
|
|
|
|
return os.path.join(MUSIC_DIR, self._file)
|
|
|
|
|
|
|
|
def set_path(self, _path: str):
|
|
|
|
self._path = _path
|
|
|
|
|
|
|
|
def get_path(self) -> str | None:
|
|
|
|
if self._path is None:
|
|
|
|
return None
|
|
|
|
return os.path.join(MUSIC_DIR, self._path)
|
|
|
|
|
|
|
|
def get_exists_on_disc(self) -> bool:
|
|
|
|
"""
|
|
|
|
returns True when file can be found on disc
|
|
|
|
returns False when file can't be found on disc or no filepath is set
|
|
|
|
"""
|
|
|
|
if not self.is_set():
|
|
|
|
return False
|
|
|
|
|
|
|
|
return os.path.exists(self.file)
|
2022-12-06 22:44:42 +00:00
|
|
|
|
2022-12-06 13:45:18 +00:00
|
|
|
def is_set(self) -> bool:
|
|
|
|
return not (self._file is None or self._path is None)
|
|
|
|
|
|
|
|
file = property(fget=get_file, fset=set_file)
|
|
|
|
path = property(fget=get_path, fset=set_path)
|
|
|
|
|
|
|
|
exists_on_disc = property(fget=get_exists_on_disc)
|
|
|
|
|
|
|
|
|
2022-12-06 22:44:42 +00:00
|
|
|
class Lyrics(DatabaseObject, SongAttribute):
|
2022-12-06 13:45:18 +00:00
|
|
|
def __init__(self, text: str, language: str, id_: str = None) -> None:
|
2022-12-06 22:44:42 +00:00
|
|
|
DatabaseObject.__init__(self, id_=id_)
|
|
|
|
SongAttribute.__init__(self)
|
2022-12-06 13:45:18 +00:00
|
|
|
self.text = text
|
|
|
|
self.language = language
|
|
|
|
|
|
|
|
|
|
|
|
class Song(DatabaseObject):
|
|
|
|
def __init__(
|
2022-12-06 22:44:42 +00:00
|
|
|
self,
|
|
|
|
id_: str = None,
|
|
|
|
mb_id: str = None,
|
|
|
|
title: str = None,
|
|
|
|
release_name: str = None,
|
|
|
|
artist_names: List[str] = [],
|
|
|
|
isrc: str = None,
|
|
|
|
length: int = None,
|
|
|
|
sources: List[Source] = None,
|
|
|
|
target: Target = None,
|
|
|
|
lyrics: List[Lyrics] = None,
|
|
|
|
metadata: dict = {},
|
|
|
|
release_ref: str = None,
|
|
|
|
artist_refs: List[Reference] = None
|
|
|
|
) -> None:
|
2022-12-06 13:45:18 +00:00
|
|
|
"""
|
|
|
|
id: is not NECESARRILY the musicbrainz id, but is DISTINCT for every song
|
|
|
|
mb_id: is the musicbrainz_id
|
|
|
|
target: Each Song can have exactly one target which can be either full or empty
|
|
|
|
lyrics: There can be multiple lyrics. Each Lyrics object can me added to multiple lyrics
|
|
|
|
"""
|
|
|
|
super().__init__(id_=id_)
|
|
|
|
# attributes
|
|
|
|
# self.id_: str | None = id_
|
|
|
|
self.mb_id: str | None = mb_id
|
|
|
|
self.title: str | None = title
|
|
|
|
self.release_name: str | None = release_name
|
|
|
|
self.isrc: str | None = isrc
|
2022-12-06 22:44:42 +00:00
|
|
|
self.length_: int | None = length
|
2022-12-06 13:45:18 +00:00
|
|
|
self.artist_names = artist_names
|
|
|
|
|
|
|
|
self.metadata = Metadata(data=metadata)
|
|
|
|
|
|
|
|
if sources is None:
|
|
|
|
sources = []
|
|
|
|
self.sources: List[Source] = sources
|
2022-12-06 22:44:42 +00:00
|
|
|
for source in self.sources:
|
|
|
|
source.add_song(self.reference)
|
|
|
|
|
2022-12-06 13:45:18 +00:00
|
|
|
if target is None:
|
|
|
|
target = Target()
|
|
|
|
self.target: Target = target
|
2022-12-06 22:44:42 +00:00
|
|
|
self.target.add_song(self.reference)
|
2022-12-06 13:45:18 +00:00
|
|
|
|
|
|
|
if lyrics is None:
|
|
|
|
lyrics = []
|
|
|
|
self.lyrics: List[Lyrics] = lyrics
|
2022-12-06 22:44:42 +00:00
|
|
|
for lyrics_ in self.lyrics:
|
|
|
|
lyrics_.add_song(self.reference)
|
2022-12-06 13:45:18 +00:00
|
|
|
|
|
|
|
self.release_ref = release_ref
|
|
|
|
self.artist_refs = artist_refs
|
|
|
|
|
|
|
|
def __str__(self) -> str:
|
2022-12-06 22:44:42 +00:00
|
|
|
return f"\"{self.title}\" by {', '.join(self.artist_names)}"
|
2022-12-06 13:45:18 +00:00
|
|
|
|
|
|
|
def __repr__(self) -> str:
|
|
|
|
return self.__str__()
|
|
|
|
|
|
|
|
def get_metadata(self):
|
|
|
|
return self.metadata.get_all_metadata()
|
|
|
|
|
|
|
|
def has_isrc(self) -> bool:
|
|
|
|
return self.isrc is not None
|
|
|
|
|
|
|
|
def get_artist_names(self) -> List[str]:
|
2022-12-06 22:44:42 +00:00
|
|
|
return self.artist_names
|
|
|
|
|
|
|
|
def get_length(self):
|
|
|
|
if self.length_ is None:
|
|
|
|
return None
|
|
|
|
return int(self.length_)
|
|
|
|
|
|
|
|
def set_length(self, length: int):
|
|
|
|
if type(length) != int:
|
|
|
|
raise TypeError(f"length of a song must be of the type int not {type(length)}")
|
|
|
|
self.length_ = length
|
|
|
|
|
|
|
|
length = property(fget=get_length, fset=set_length)
|
|
|
|
|
2022-12-06 13:45:18 +00:00
|
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
|
"""
|
|
|
|
Example for creating a Song object
|
|
|
|
"""
|
|
|
|
|
|
|
|
song = Song(
|
|
|
|
title="Vein Deep in the Solution",
|
|
|
|
release_name="One Final Action",
|
|
|
|
target=Target(file="~/Music/genre/artist/album/song.mp3", path="~/Music/genre/artist/album"),
|
|
|
|
metadata={
|
|
|
|
"album": "One Final Action"
|
|
|
|
},
|
|
|
|
lyrics=[
|
|
|
|
Lyrics(text="these are some depressive lyrics", language="en")
|
|
|
|
],
|
|
|
|
sources=[
|
|
|
|
Source(src="youtube", url="https://youtu.be/dfnsdajlhkjhsd")
|
|
|
|
]
|
|
|
|
)
|