pythonasyncioapiasyncfastapiframeworkjsonjson-schemaopenapiopenapi3pydanticpython-typespython3redocreststarletteswaggerswagger-uiuvicornweb
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
24 lines
662 B
24 lines
662 B
from contextvars import ContextVar
|
|
|
|
import peewee
|
|
|
|
DATABASE_NAME = "test.db"
|
|
db_state_default = {"closed": None, "conn": None, "ctx": None, "transactions": None}
|
|
db_state = ContextVar("db_state", default=db_state_default.copy())
|
|
|
|
|
|
class PeeweeConnectionState(peewee._ConnectionState):
|
|
def __init__(self, **kwargs):
|
|
super().__setattr__("_state", db_state)
|
|
super().__init__(**kwargs)
|
|
|
|
def __setattr__(self, name, value):
|
|
self._state.get()[name] = value
|
|
|
|
def __getattr__(self, name):
|
|
return self._state.get()[name]
|
|
|
|
|
|
db = peewee.SqliteDatabase(DATABASE_NAME, check_same_thread=False)
|
|
|
|
db._state = PeeweeConnectionState()
|
|
|