diff --git a/app/admin_manager.py b/app/admin_manager.py index d36947c..8323b01 100644 --- a/app/admin_manager.py +++ b/app/admin_manager.py @@ -2,13 +2,10 @@ from fastapi import FastAPI from sqladmin import Admin from app.database.session import engine -from app.lyric.api.lyrics_admin import ( - LyricsAttributeAdmin, - LyricsPromptTemplateAdmin, - LyricsSongResultsAllAdmin, - LyricsSongSampleAdmin, - LyricsStoreDefaultInfoAdmin, -) +from app.home.api.home_admin import ImageAdmin, ProjectAdmin +from app.lyric.api.lyrics_admin import LyricAdmin +from app.song.api.song_admin import SongAdmin +from app.video.api.video_admin import VideoAdmin from config import prj_settings # https://github.com/aminalaee/sqladmin @@ -25,10 +22,17 @@ def init_admin( base_url=base_url, ) - admin.add_view(LyricsStoreDefaultInfoAdmin) - admin.add_view(LyricsAttributeAdmin) - admin.add_view(LyricsSongSampleAdmin) - admin.add_view(LyricsPromptTemplateAdmin) - admin.add_view(LyricsSongResultsAllAdmin) + # 프로젝트 관리 + admin.add_view(ProjectAdmin) + admin.add_view(ImageAdmin) + + # 가사 관리 + admin.add_view(LyricAdmin) + + # 노래 관리 + admin.add_view(SongAdmin) + + # 영상 관리 + admin.add_view(VideoAdmin) return admin diff --git a/app/database/session.py b/app/database/session.py index 3774ab8..b081725 100644 --- a/app/database/session.py +++ b/app/database/session.py @@ -37,22 +37,19 @@ AsyncSessionLocal = async_sessionmaker( async def create_db_tables(): - async with engine.begin() as connection: - from app.lyric.models import ( # noqa: F401 - Attribute, - PromptTemplate, - SongResultsAll, - SongSample, - StoreDefaultInfo, - ) + import asyncio - print("Creating database tables...") + # 모델 import (테이블 메타데이터 등록용) + from app.home.models import Image, Project # noqa: F401 + from app.lyric.models import Lyric # noqa: F401 + from app.song.models import Song # noqa: F401 + from app.video.models import Video # noqa: F401 - import asyncio + print("Creating database tables...") - async with asyncio.timeout(10): - async with engine.begin() as connection: - await connection.run_sync(Base.metadata.create_all) + async with asyncio.timeout(10): + async with engine.begin() as connection: + await connection.run_sync(Base.metadata.create_all) # FastAPI 의존성용 세션 제너레이터