Spaces:

ashfortune
/

nexus-ai-api

Runtime error

github-actions[bot]

Auto Deploy from Monorepo: Merge pull request #72 from human13th2team/dev

df5e016 6 months ago

945 Bytes

import os

from dotenv import load_dotenv
from sqlalchemy.ext.asyncio import AsyncSession, async_sessionmaker, create_async_engine
from sqlalchemy.orm import DeclarativeBase

# .env 파일 로드
load_dotenv()

DATABASE_URL = os.getenv("DATABASE_URL")

if not DATABASE_URL:
    raise ValueError("DATABASE_URL이 설정되지 않았습니다. .env 파일을 확인해주세요.")

# 비동기 엔진 생성
engine = create_async_engine(DATABASE_URL, echo=True, connect_args={"statement_cache_size": 0})

# 비동기 세션 메이커
AsyncSessionLocal = async_sessionmaker(
    bind=engine,
    class_=AsyncSession,
    expire_on_commit=False,
)


# Base 클래스 정의 (SQLAlchemy 2.0 style)
class Base(DeclarativeBase):
    pass


# DB 세션 의존성 주입을 위한 제너레이터
async def get_db():
    async with AsyncSessionLocal() as session:
        try:
            yield session
        finally:
            await session.close()

	import os

	from dotenv import load_dotenv
	from sqlalchemy.ext.asyncio import AsyncSession, async_sessionmaker, create_async_engine
	from sqlalchemy.orm import DeclarativeBase

	# .env 파일 로드
	load_dotenv()

	DATABASE_URL = os.getenv("DATABASE_URL")

	if not DATABASE_URL:
	raise ValueError("DATABASE_URL이 설정되지 않았습니다. .env 파일을 확인해주세요.")

	# 비동기 엔진 생성
	engine = create_async_engine(DATABASE_URL, echo=True, connect_args={"statement_cache_size": 0})

	# 비동기 세션 메이커
	AsyncSessionLocal = async_sessionmaker(
	bind=engine,
	class_=AsyncSession,
	expire_on_commit=False,
	)


	# Base 클래스 정의 (SQLAlchemy 2.0 style)
	class Base(DeclarativeBase):
	pass


	# DB 세션 의존성 주입을 위한 제너레이터
	async def get_db():
	async with AsyncSessionLocal() as session:
	try:
	yield session
	finally:
	await session.close()