Coverage for app/core/db.py: 80%

1from sqlmodel import Session, create_engine, select (empty)

2 

3from app import crud (empty)

4from app.core.config import settings (empty)

5from app.models import User, UserCreate (empty)

6 

7engine = create_engine(str(settings.DATABASE_URL), pool_pre_ping=True) (empty)

8 

9 

10# make sure all SQLModel models are imported (app.models) before initializing DB 

11# otherwise, SQLModel might fail to initialize relationships properly 

12# for more details: https://github.com/fastapi/full-stack-fastapi-template/issues/28 

13 

14 

15def init_db(session: Session) -> None: (empty)

16 # Tables should be created with Alembic migrations 

17 # But if you don't want to use migrations, create 

18 # the tables un-commenting the next lines 

19 # from sqlmodel import SQLModel 

20 

21 # This works because the models are already imported and registered from app.models 

22 # SQLModel.metadata.create_all(engine) 

23 

24 user = session.exec( (empty)

25 select(User).where(User.email == settings.FIRST_SUPERUSER) 

26 ).first() 

27 if not user: (empty)

28 user_in = UserCreate( 

29 email=settings.FIRST_SUPERUSER, 

30 password=settings.FIRST_SUPERUSER_PASSWORD, 

31 is_superuser=True, 

32 ) 

33 user = crud.create_user(session=session, user_create=user_in)