Mercurial

comparison love/poppy/utils/database.py @ 38:cf9caa4abc3e

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
[Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations.
author MrJuneJune <me@mrjunejune.com>
date Mon, 01 Dec 2025 20:35:56 -0800
parents
children
comparison
equal deleted inserted replaced
37:fb9bcd3145cb 38:cf9caa4abc3e
1 from sqlmodel import SQLModel, create_engine
2 from sqlalchemy.ext.asyncio import (
3 create_async_engine,
4 async_sessionmaker,
5 AsyncEngine,
6 AsyncSession,
7 )
8 import os
9
10 ASYNC_DATABASE_URL = os.getenv("ASYNC_DATABASE_URL", "sqlite+aiosqlite:///./poppy.db")
11 SYNC_DATABASE_URL = os.getenv("SYNC_DATABASE_URL", "sqlite:///./poppy.db")
12
13 # For migrations, we can probably use async?
14 sync_engine = create_engine(
15 SYNC_DATABASE_URL,
16 echo=False,
17 future=True,
18 )
19
20 async_engine: AsyncEngine = create_async_engine(
21 ASYNC_DATABASE_URL,
22 echo=False,
23 future=True,
24 )
25
26 AsyncSessionLocal = async_sessionmaker(
27 bind=async_engine,
28 class_=AsyncSession,
29 expire_on_commit=False,
30 )
31
32
33 async def get_session():
34 async with AsyncSessionLocal() as session:
35 yield session
36
37
38 def create_db_and_tables():
39 SQLModel.metadata.create_all(sync_engine)