61 lines
1.4 KiB
Python
61 lines
1.4 KiB
Python
from typing import TYPE_CHECKING
|
|
|
|
from pydantic import BaseSettings
|
|
from sqlalchemy import create_engine
|
|
from sqlmodel import SQLModel, Session
|
|
|
|
from learn_sql_model.standard_config import load
|
|
|
|
if TYPE_CHECKING:
|
|
from sqlalchemy import Engine
|
|
|
|
|
|
class Database:
|
|
def __init__(self, config: "Config" = None) -> None:
|
|
if config is None:
|
|
|
|
self.config = get_config()
|
|
else:
|
|
self.config = config
|
|
self.create_db_and_tables()
|
|
|
|
@property
|
|
def engine(self) -> "Engine":
|
|
return create_engine(self.config.database_url)
|
|
|
|
def session(self) -> "Session":
|
|
return Session(self.engine)
|
|
|
|
def create_db_and_tables(self) -> None:
|
|
from learn_sql_model.models.hero import Hero
|
|
from learn_sql_model.models.pet import Pet
|
|
|
|
__all__ = [Hero, Pet]
|
|
|
|
SQLModel.metadata.create_all(self.engine)
|
|
|
|
|
|
class Config(BaseSettings):
|
|
database_url: str = "sqlite:///database.db"
|
|
port: int = 5000
|
|
|
|
class Config:
|
|
env_prefix = "LEARN_SQL_MODEL_"
|
|
|
|
@property
|
|
def database(self) -> Database:
|
|
return get_database(config=self)
|
|
|
|
|
|
def get_database(config: Config = None) -> Database:
|
|
|
|
if config is None:
|
|
config = get_config()
|
|
|
|
return Database(config)
|
|
|
|
|
|
def get_config(overrides: dict = {}) -> Config:
|
|
raw_config = load("learn_sql_model")
|
|
config = Config(**raw_config, **overrides)
|
|
return config
|