added album status table and migrations

audiostats/application/__init__.py

@@ -0,0 +1 @@
+from .models import Status, Success

audiostats/application/dto_mappers.py

@@ -2,19 +2,6 @@
 
 from audiostats.handlers import AlbumDTO, TrackDTO
 
-
-# def create_album_f_dto(album : AlbumDTO) -> Album:
-#     return Album(title=album.title,
-#                 performer=album.performer,
-#                 year=album.year,
-#                 path=album.path,
-#                 cover=album.cover,
-#                 tracks=[Track(title=track.title,
-#                               number=track.number,
-#                               path=track.path,
-#                               offset=track.offset,
-#                               duration=track.duration) for track in album.tracks])
-
 def update_track_orm_f_dto(old : Track, new : TrackDTO) -> None:
     old.title = new.title
     old.number = new.number

audiostats/application/models.py

@@ -0,0 +1,11 @@
+from enum import StrEnum
+
+class Status(StrEnum):
+    ADDED = 'added'
+    MODIFIED = 'modified'
+
+class Success(StrEnum):
+    SUCCESS = 'success'
+    WARNING = 'warning'
+    ERROR = 'error'
+

audiostats/db/models.py

@@ -1,8 +1,10 @@
 from datetime import datetime
 
-from sqlalchemy import String, Integer, UniqueConstraint, ForeignKey, Float, DATETIME
+from sqlalchemy import String, Integer, UniqueConstraint, ForeignKey, Float, Enum, DateTime
 from sqlalchemy.orm import declarative_base, Mapped, mapped_column, relationship
 
+from audiostats.application import Status, Success
+
 Base = declarative_base()
 
 MAX_PATH_FIELD_LEN = 200
@@ -34,7 +36,8 @@ class Album(Base):
     year : Mapped[int | None] = mapped_column(Integer, nullable=True)
     path : Mapped[str | None] = mapped_column(String(MAX_PATH_FIELD_LEN), nullable=True)
     cover : Mapped[str | None] = mapped_column(String(MAX_PATH_FIELD_LEN), nullable=True)
-    tracks : Mapped[list['Track']] = relationship('Track', back_populates='album')
+    tracks : Mapped[list['Track']] = relationship('Track', back_populates='album', lazy='noload')
+    album_statuses : Mapped[list['AlbumStatus']] = relationship('AlbumStatus', back_populates='album', lazy='noload')
 
     def __repr__(self):
         return f'<Album(year={self.year}, performer={self.performer}, title={self.title})>'
@@ -63,7 +66,7 @@ class Track(Base):
     path : Mapped[str] = mapped_column(String(MAX_PATH_FIELD_LEN), nullable=True)
     offset : Mapped[float | None] = mapped_column(Float, nullable=True)
     duration : Mapped[float | None] = mapped_column(Float, nullable=True)
-    album : Mapped["Album"]= relationship('Album', back_populates='tracks')
+    album : Mapped["Album"]= relationship('Album', back_populates='tracks', lazy='noload')
 
     def __repr__(self):
         return f'<Track(title={self.title}, album_id={self.album_id}, number={self.number})>'
@@ -72,15 +75,18 @@ def __str__(self):
         return f'{self.number} - {self.title}'
 
 
-# class AlbumStatus(Base):
-#     """Represents **album_statuses** line as orm object"""
-#
-#     __tablename__ = 'album_statuses'
-#
-#     id : Mapped[int] = mapped_column(Integer, primary_key=True)
-#     album_id : Mapped[int] = mapped_column(Integer, ForeignKey('albums.id', ondelete='CASCADE'), index=True)
-#     time_stamp : Mapped[datetime] = mapped_column(DATETIME, nullable=False)
-#     status : Mapped[]
-#     success : Mapped[]
+class AlbumStatus(Base):
+    """Represents **album_statuses** line as orm object"""
+
+    __tablename__ = 'album_statuses'
+
+    id : Mapped[int] = mapped_column(Integer, primary_key=True)
+    album_id : Mapped[int] = mapped_column(Integer, ForeignKey('albums.id', ondelete='CASCADE'), index=True)
+    time_stamp : Mapped[datetime] = mapped_column(DateTime, nullable=False)
+    status : Mapped[Status] = mapped_column(Enum(Status), nullable=False)
+    success : Mapped[Success] = mapped_column(Enum(Success), nullable=False)
+    album : Mapped["Album"] = relationship('Album', back_populates='album_statuses', lazy='noload')
+
+
 
 

audiostats/db/repositories.py

@@ -3,7 +3,7 @@
 from sqlalchemy.ext.asyncio import AsyncSession
 from sqlalchemy.orm import selectinload
 
-from .models import Album
+from .models import Album, AlbumStatus
 from audiostats.handlers import AlbumDTO
 
 from audiostats.application.dto_mappers import create_album_dto_f_orm, update_album_orm_meta_f_dto, update_track_orm_f_dto, create_track_orm_f_dto
@@ -17,10 +17,12 @@ def __init__(self, session : AsyncSession):
 
     async def upsert(self, album_data : AlbumDTO):
         album = await self.find_by_title_performer(album_data.title, album_data.performer)
+        # album_status = AlbumStatus()
 
         if not album:
             album = Album()
             self._session.add(album)
+
         update_album_orm_meta_f_dto(album, album_data)
 
         old_tracks_by_title = {track.title: track for track in album.tracks}

migrations/versions/a337b3f88bf1_added_album_status.py

@@ -0,0 +1,47 @@
+"""added album status
+
+Revision ID: a337b3f88bf1
+Revises: bd8d699fa622
+Create Date: 2025-09-16 20:08:54.362324
+
+"""
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision: str = 'a337b3f88bf1'
+down_revision: Union[str, Sequence[str], None] = 'bd8d699fa622'
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    """Upgrade schema."""
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.create_table('album_statuses',
+    sa.Column('id', sa.Integer(), nullable=False),
+    sa.Column('album_id', sa.Integer(), nullable=False),
+    sa.Column('time_stamp', sa.DateTime(), nullable=False),
+    sa.Column('status', sa.Enum('ADDED', 'MODIFIED', name='status'), nullable=False),
+    sa.Column('success', sa.Enum('SUCCESS', 'WARNING', 'ERROR', name='success'), nullable=False),
+    sa.ForeignKeyConstraint(['album_id'], ['albums.id'], ondelete='CASCADE'),
+    sa.PrimaryKeyConstraint('id')
+    )
+    op.create_index(op.f('ix_album_statuses_album_id'), 'album_statuses', ['album_id'], unique=False)
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    """Downgrade schema."""
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_index(op.f('ix_album_statuses_album_id'), table_name='album_statuses')
+    op.drop_table('album_statuses')
+
+    status_enum = sa.Enum('ADDED', 'MODIFIED', name='status')
+    success_enum = sa.Enum('SUCCESS', 'WARNING', 'ERROR', name='success')
+    status_enum.drop(op.get_bind())
+    success_enum.drop(op.get_bind())
+    # ### end Alembic commands ###