Spaces:

Marintosti
/

deploy-machine-learning

Runtime error

deploy-machine-learning / src /config /db.py

marintosti12

feat(test) : add some fonctionals tets

4ba8e3d about 1 year ago

615 Bytes

from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker, DeclarativeBase
from pydantic_settings import BaseSettings, SettingsConfigDict

class Settings(BaseSettings):
    DATABASE_URL: str
    model_config = SettingsConfigDict(env_file=".env", env_file_encoding="utf-8", extra="ignore",)

settings = Settings()

class Base(DeclarativeBase):
    pass

engine = create_engine(settings.DATABASE_URL, echo=True, future=True)
SessionLocal = sessionmaker(bind=engine, autocommit=False, autoflush=False)

def get_db():
    db = SessionLocal()
    try:
        yield db
    finally:
        db.close()

	from sqlalchemy import create_engine
	from sqlalchemy.orm import sessionmaker, DeclarativeBase
	from pydantic_settings import BaseSettings, SettingsConfigDict

	class Settings(BaseSettings):
	DATABASE_URL: str
	model_config = SettingsConfigDict(env_file=".env", env_file_encoding="utf-8", extra="ignore",)

	settings = Settings()

	class Base(DeclarativeBase):
	pass

	engine = create_engine(settings.DATABASE_URL, echo=True, future=True)
	SessionLocal = sessionmaker(bind=engine, autocommit=False, autoflush=False)

	def get_db():
	db = SessionLocal()
	try:
	yield db
	finally:
	db.close()