eeeeeeeeeeee

This commit is contained in:
Hellow2 2023-02-16 17:24:22 +01:00
parent a56233d5a6
commit 019c24e059
2 changed files with 130 additions and 8 deletions

View File

@ -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)

View File

@ -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
""" """
pass 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