feat: Add Extraction model and seed main playlist functionality

This commit is contained in:
JSC
2025-07-28 19:39:32 +02:00
parent 34e6289f92
commit c993230f98
8 changed files with 121 additions and 49 deletions

View File

@@ -5,6 +5,7 @@ from sqlmodel.ext.asyncio.session import AsyncSession
from app.core.logging import get_logger
from app.models.plan import Plan
from app.models.playlist import Playlist
logger = get_logger(__name__)
@@ -52,12 +53,37 @@ async def seed_plans(session: AsyncSession) -> None:
logger.info("Successfully seeded %d plans", len(initial_plans))
async def seed_main_playlist(session: AsyncSession) -> None:
"""Seed the main playlist with initial data."""
logger.info("Seeding main playlist data")
# Check if the main playlist already exists
existing_playlist = await session.exec(select(Playlist).where(Playlist.is_main))
if existing_playlist.first():
logger.info("Main playlist already exists, skipping seeding")
return
# Create the main playlist
main_playlist = Playlist(
name="All",
description="The default main playlist with all the tracks",
is_main=True,
is_deletable=False,
is_current=True,
)
session.add(main_playlist)
await session.commit()
logger.info("Successfully seeded main playlist")
async def seed_all_data(session: AsyncSession) -> None:
"""Seed all initial data."""
logger.info("Starting data seeding")
try:
await seed_plans(session)
await seed_main_playlist(session)
logger.info("Data seeding completed successfully")
except Exception:
logger.exception("Failed to seed data")