FilesExpand file tree

 master

/

database.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

24 lines (15 loc) · 662 Bytes

 master

/

database.py

Copy path

Top

File metadata and controls

• Code
• Blame

24 lines (15 loc) · 662 Bytes

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

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()