35 lines
807 B
Python
35 lines
807 B
Python
from fastapi import FastAPI
|
|
from sqladmin import Admin
|
|
|
|
from app.database.session import engine
|
|
from app.lyrics.api.lyrics_admin import (
|
|
LyricsAttributeAdmin,
|
|
LyricsPromptTemplateAdmin,
|
|
LyricsSongResultsAllAdmin,
|
|
LyricsSongSampleAdmin,
|
|
LyricsStoreDefaultInfoAdmin,
|
|
)
|
|
from config import prj_settings
|
|
|
|
# https://github.com/aminalaee/sqladmin
|
|
|
|
|
|
def init_admin(
|
|
app: FastAPI,
|
|
db_engine: engine,
|
|
base_url: str = prj_settings.ADMIN_BASE_URL,
|
|
) -> Admin:
|
|
admin = Admin(
|
|
app,
|
|
db_engine,
|
|
base_url=base_url,
|
|
)
|
|
|
|
admin.add_view(LyricsStoreDefaultInfoAdmin)
|
|
admin.add_view(LyricsAttributeAdmin)
|
|
admin.add_view(LyricsSongSampleAdmin)
|
|
admin.add_view(LyricsPromptTemplateAdmin)
|
|
admin.add_view(LyricsSongResultsAllAdmin)
|
|
|
|
return admin
|