add db config at startup
This commit is contained in:
parent
922206b365
commit
b47ed30681
|
@ -52,6 +52,25 @@ def bundle_vendored():
|
|||
with open(outputpath, "w") as f:
|
||||
f.write(output)
|
||||
|
||||
async def get_admin_settings():
|
||||
from lnbits.extensions.admin.models import Admin
|
||||
|
||||
async with core_db.connect() as conn:
|
||||
|
||||
if conn.type == SQLITE:
|
||||
exists = await conn.fetchone(
|
||||
"SELECT * FROM sqlite_master WHERE type='table' AND name='admin'"
|
||||
)
|
||||
elif conn.type in {POSTGRES, COCKROACH}:
|
||||
exists = await conn.fetchone(
|
||||
"SELECT * FROM information_schema.tables WHERE table_name = 'admin'"
|
||||
)
|
||||
if not exists:
|
||||
return False
|
||||
|
||||
row = await conn.fetchone("SELECT * from admin")
|
||||
|
||||
return Admin(**row) if row else None
|
||||
|
||||
async def migrate_databases():
|
||||
"""Creates the necessary databases if they don't exist already; or migrates them."""
|
||||
|
|
Loading…
Reference in New Issue
Block a user