eeeeeeeeeeee
This commit is contained in:
parent
a56233d5a6
commit
019c24e059
@ -72,9 +72,9 @@ class Song(BaseModel):
|
|||||||
|
|
||||||
class Source(BaseModel):
|
class Source(BaseModel):
|
||||||
"""A class representing a source of a song in the music database."""
|
"""A class representing a source of a song in the music database."""
|
||||||
|
ContentTypes = Union[Song, Album, Artist]
|
||||||
|
|
||||||
type: str = CharField()
|
page: str = CharField()
|
||||||
src: str = CharField()
|
|
||||||
url: str = CharField()
|
url: str = CharField()
|
||||||
|
|
||||||
content_type: str = CharField()
|
content_type: str = CharField()
|
||||||
@ -82,7 +82,7 @@ class Source(BaseModel):
|
|||||||
content: ForeignKeyField = ForeignKeyField('self', backref='content_items', null=True)
|
content: ForeignKeyField = ForeignKeyField('self', backref='content_items', null=True)
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def content_object(self) -> Union[Song, Album, Artist, None]:
|
def content_object(self) -> Union[Song, Album, Artist]:
|
||||||
"""Get the content associated with the source as an object."""
|
"""Get the content associated with the source as an object."""
|
||||||
if self.content_type == 'Song':
|
if self.content_type == 'Song':
|
||||||
return Song.get(Song.id == self.content_id)
|
return Song.get(Song.id == self.content_id)
|
||||||
|
@ -1,4 +1,5 @@
|
|||||||
from typing import Union
|
from typing import Union, Set
|
||||||
|
import traceback
|
||||||
from peewee import (
|
from peewee import (
|
||||||
SqliteDatabase,
|
SqliteDatabase,
|
||||||
MySQLDatabase,
|
MySQLDatabase,
|
||||||
@ -6,14 +7,135 @@ from peewee import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
from . import objects
|
from . import objects
|
||||||
|
from . import data_models
|
||||||
|
|
||||||
# just a Type for type hintung. You can't do anything with it.
|
# just a Type for type hintung. You can't do anything with it.
|
||||||
Database = Union[SqliteDatabase, PostgresqlDatabase, MySQLDatabase]
|
Database = Union[SqliteDatabase, PostgresqlDatabase, MySQLDatabase]
|
||||||
|
|
||||||
|
|
||||||
class Session:
|
class Session:
|
||||||
"""
|
"""
|
||||||
do the thingie with the with keyword
|
Context manager for a database session
|
||||||
overload __end__ and maybe __start__ i dunfckinnow
|
|
||||||
|
Usage:
|
||||||
|
with Session(database) as session:
|
||||||
|
# Perform database operations using session object
|
||||||
|
|
||||||
|
Args:
|
||||||
|
database: An instance of a database connection from Peewee
|
||||||
|
|
||||||
|
Attributes:
|
||||||
|
database: An instance of a database connection from Peewee
|
||||||
|
"""
|
||||||
|
def __init__(self, database: Database) -> None:
|
||||||
|
"""
|
||||||
|
Initialize a database session
|
||||||
|
|
||||||
|
Args:
|
||||||
|
database: An instance of a database connection from Peewee
|
||||||
|
"""
|
||||||
|
self.database = database
|
||||||
|
|
||||||
|
self.added_song_ids: Set[str] = set()
|
||||||
|
self.added_album_ids: Set[str] = set()
|
||||||
|
self.added_artist_ids: Set[str] = set()
|
||||||
|
|
||||||
|
def __enter__(self, database: Database):
|
||||||
|
"""
|
||||||
|
Enter the context of the database session
|
||||||
|
|
||||||
|
Args:
|
||||||
|
database: An instance of a database connection from Peewee
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
self: The instance of the session object
|
||||||
|
"""
|
||||||
|
print('Entering the context')
|
||||||
|
self.__init__(database=database)
|
||||||
|
return self
|
||||||
|
|
||||||
|
def __exit__(self, exc_type, exc_val, exc_tb):
|
||||||
|
"""
|
||||||
|
Exit the context of the database session
|
||||||
|
|
||||||
|
Args:
|
||||||
|
exc_type: The type of the raised exception
|
||||||
|
exc_val: The value of the raised exception
|
||||||
|
exc_tb: The traceback of the raised exception
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
bool: True if no exception was raised, False otherwise
|
||||||
|
"""
|
||||||
|
if exc_val is not None:
|
||||||
|
traceback.print_tb(exc_tb)
|
||||||
|
print(f"Exception of type {exc_type} occurred with message: {exc_val}")
|
||||||
|
|
||||||
|
self.commit()
|
||||||
|
return exc_val is None
|
||||||
|
|
||||||
|
def add_source(self, source: objects.Source, connected_to: data_models.Source.ContentTypes) -> data_models.Source:
|
||||||
|
db_source = data_models.Source(
|
||||||
|
page=source.page_str,
|
||||||
|
url=source.url,
|
||||||
|
content_object=connected_to
|
||||||
|
).use(self.database)
|
||||||
|
|
||||||
|
return db_source
|
||||||
|
|
||||||
|
def add_song(self, song: objects.Song) -> data_models.Song:
|
||||||
|
"""
|
||||||
|
Add a song object to the session
|
||||||
|
|
||||||
|
Args:
|
||||||
|
song: An instance of the Song object
|
||||||
|
"""
|
||||||
|
if song.dynamic:
|
||||||
|
return
|
||||||
|
if song.id in self.added_song_ids:
|
||||||
|
return
|
||||||
|
self.added_song_ids.add(song.id)
|
||||||
|
|
||||||
|
db_song: data_models.Song = data_models.Song(
|
||||||
|
id=song.id,
|
||||||
|
name=song.title,
|
||||||
|
isrc=song.isrc,
|
||||||
|
length=song.length,
|
||||||
|
tracksort=song.tracksort,
|
||||||
|
genre=song.genre
|
||||||
|
).use(self.database)
|
||||||
|
|
||||||
|
for source in song.source_list:
|
||||||
|
self.add_source(source=source, connected_to=db_song)
|
||||||
|
|
||||||
|
return db_song
|
||||||
|
|
||||||
|
def add_album(self, album: objects.Album):
|
||||||
|
"""
|
||||||
|
Add an album object to the session
|
||||||
|
|
||||||
|
Args:
|
||||||
|
album: An instance of the Album object
|
||||||
|
"""
|
||||||
|
if album.dynamic:
|
||||||
|
return
|
||||||
|
if album.id in self.added_album_ids:
|
||||||
|
return
|
||||||
|
self.added_album_ids.add(album.id)
|
||||||
|
|
||||||
|
def add_artist(self, artist: objects.Artist):
|
||||||
|
"""
|
||||||
|
Add an artist object to the session
|
||||||
|
|
||||||
|
Args:
|
||||||
|
artist: An instance of the Artist object
|
||||||
|
"""
|
||||||
|
if artist.dynamic:
|
||||||
|
return
|
||||||
|
if artist.id in self.added_artist_ids:
|
||||||
|
return
|
||||||
|
self.added_artist_ids.add(artist.id)
|
||||||
|
|
||||||
|
def commit(self):
|
||||||
|
"""
|
||||||
|
Commit changes to the database
|
||||||
"""
|
"""
|
||||||
pass
|
pass
|
Loading…
Reference in New Issue
Block a user