aiteamutils 0.2.13__py3-none-any.whl → 0.2.14__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- aiteamutils/database.py +37 -2
- aiteamutils/dependencies.py +4 -38
- aiteamutils/version.py +1 -1
- {aiteamutils-0.2.13.dist-info → aiteamutils-0.2.14.dist-info}/METADATA +1 -1
- {aiteamutils-0.2.13.dist-info → aiteamutils-0.2.14.dist-info}/RECORD +6 -6
- {aiteamutils-0.2.13.dist-info → aiteamutils-0.2.14.dist-info}/WHEEL +0 -0
aiteamutils/database.py
CHANGED
@@ -6,17 +6,52 @@ from sqlalchemy.exc import IntegrityError, SQLAlchemyError
|
|
6
6
|
from sqlalchemy.pool import QueuePool
|
7
7
|
from contextlib import asynccontextmanager
|
8
8
|
from sqlalchemy import or_
|
9
|
-
from fastapi import Request
|
9
|
+
from fastapi import Request, Depends
|
10
10
|
from ulid import ULID
|
11
11
|
from sqlalchemy.sql import Select
|
12
12
|
|
13
13
|
from .exceptions import ErrorCode, CustomException
|
14
14
|
from .base_model import Base, BaseColumn
|
15
15
|
from .enums import ActivityType
|
16
|
-
from .config import settings
|
17
16
|
|
18
17
|
T = TypeVar("T", bound=BaseColumn)
|
19
18
|
|
19
|
+
# 전역 데이터베이스 서비스 인스턴스
|
20
|
+
_database_service: 'DatabaseService' | None = None
|
21
|
+
|
22
|
+
def get_database_service() -> 'DatabaseService':
|
23
|
+
"""DatabaseService 인스턴스를 반환하는 함수
|
24
|
+
|
25
|
+
Returns:
|
26
|
+
DatabaseService: DatabaseService 인스턴스
|
27
|
+
|
28
|
+
Raises:
|
29
|
+
RuntimeError: DatabaseService가 초기화되지 않은 경우
|
30
|
+
"""
|
31
|
+
if _database_service is None:
|
32
|
+
raise RuntimeError("DatabaseService not initialized. Call init_settings with db_url first.")
|
33
|
+
return _database_service
|
34
|
+
|
35
|
+
async def get_db() -> AsyncGenerator[AsyncSession, None]:
|
36
|
+
"""데이터베이스 세션을 생성하고 반환하는 비동기 제너레이터."""
|
37
|
+
db_service = get_database_service()
|
38
|
+
async with db_service.get_session() as session:
|
39
|
+
try:
|
40
|
+
yield session
|
41
|
+
finally:
|
42
|
+
await session.close()
|
43
|
+
|
44
|
+
def get_database_session(db: AsyncSession = Depends(get_db)) -> 'DatabaseService':
|
45
|
+
"""DatabaseService 의존성
|
46
|
+
|
47
|
+
Args:
|
48
|
+
db (AsyncSession): 데이터베이스 세션
|
49
|
+
|
50
|
+
Returns:
|
51
|
+
DatabaseService: DatabaseService 인스턴스
|
52
|
+
"""
|
53
|
+
return DatabaseService(session=db)
|
54
|
+
|
20
55
|
class DatabaseService:
|
21
56
|
def __init__(
|
22
57
|
self,
|
aiteamutils/dependencies.py
CHANGED
@@ -1,10 +1,9 @@
|
|
1
|
-
from typing import Type, Dict, Tuple, Any,
|
2
|
-
from sqlalchemy.ext.asyncio import AsyncSession
|
1
|
+
from typing import Type, Dict, Tuple, Any, Callable
|
3
2
|
from fastapi import Depends, status
|
4
3
|
from fastapi.security import OAuth2PasswordBearer
|
5
4
|
from jose import JWTError, jwt
|
6
5
|
|
7
|
-
from .database import DatabaseService
|
6
|
+
from .database import DatabaseService, get_database_service
|
8
7
|
from .exceptions import CustomException, ErrorCode
|
9
8
|
|
10
9
|
class Settings:
|
@@ -14,7 +13,6 @@ class Settings:
|
|
14
13
|
self.JWT_ALGORITHM = jwt_algorithm
|
15
14
|
|
16
15
|
_settings: Settings | None = None
|
17
|
-
_database_service: DatabaseService | None = None
|
18
16
|
|
19
17
|
def init_settings(
|
20
18
|
jwt_secret: str,
|
@@ -38,10 +36,11 @@ def init_settings(
|
|
38
36
|
db_pool_timeout (int, optional): 커넥션 풀 타임아웃
|
39
37
|
db_pool_recycle (int, optional): 커넥션 재활용 시간
|
40
38
|
"""
|
41
|
-
global _settings
|
39
|
+
global _settings
|
42
40
|
_settings = Settings(jwt_secret, jwt_algorithm)
|
43
41
|
|
44
42
|
if db_url:
|
43
|
+
from .database import _database_service
|
45
44
|
_database_service = DatabaseService(
|
46
45
|
db_url=db_url,
|
47
46
|
db_echo=db_echo,
|
@@ -64,39 +63,6 @@ def get_settings() -> Settings:
|
|
64
63
|
raise RuntimeError("Settings not initialized. Call init_settings first.")
|
65
64
|
return _settings
|
66
65
|
|
67
|
-
def get_database_service() -> DatabaseService:
|
68
|
-
"""DatabaseService 인스턴스를 반환하는 함수
|
69
|
-
|
70
|
-
Returns:
|
71
|
-
DatabaseService: DatabaseService 인스턴스
|
72
|
-
|
73
|
-
Raises:
|
74
|
-
RuntimeError: DatabaseService가 초기화되지 않은 경우
|
75
|
-
"""
|
76
|
-
if _database_service is None:
|
77
|
-
raise RuntimeError("DatabaseService not initialized. Call init_settings with db_url first.")
|
78
|
-
return _database_service
|
79
|
-
|
80
|
-
async def get_db() -> AsyncGenerator[AsyncSession, None]:
|
81
|
-
"""데이터베이스 세션을 생성하고 반환하는 비동기 제너레이터."""
|
82
|
-
db_service = get_database_service()
|
83
|
-
async with db_service.get_session() as session:
|
84
|
-
try:
|
85
|
-
yield session
|
86
|
-
finally:
|
87
|
-
await session.close()
|
88
|
-
|
89
|
-
def get_database_session(db: AsyncSession = Depends(get_db)) -> DatabaseService:
|
90
|
-
"""DatabaseService 의존성
|
91
|
-
|
92
|
-
Args:
|
93
|
-
db (AsyncSession): 데이터베이스 세션
|
94
|
-
|
95
|
-
Returns:
|
96
|
-
DatabaseService: DatabaseService 인스턴스
|
97
|
-
"""
|
98
|
-
return DatabaseService(session=db)
|
99
|
-
|
100
66
|
class ServiceRegistry:
|
101
67
|
"""서비스 레지스트리를 관리하는 클래스"""
|
102
68
|
def __init__(self):
|
aiteamutils/version.py
CHANGED
@@ -1,2 +1,2 @@
|
|
1
1
|
"""버전 정보"""
|
2
|
-
__version__ = "0.2.
|
2
|
+
__version__ = "0.2.14"
|
@@ -4,13 +4,13 @@ aiteamutils/base_repository.py,sha256=qdwQ7Sj2fUqxpDg6cWM48n_QbwPK_VUlG9zTSem8iC
|
|
4
4
|
aiteamutils/base_service.py,sha256=E4dHGE0DvhmRyFplh46SwKJOSF_nUL7OAsCkf_ZJF_8,24733
|
5
5
|
aiteamutils/cache.py,sha256=tr0Yn8VPYA9QHiKCUzciVlQ2J1RAwNo2K9lGMH4rY3s,1334
|
6
6
|
aiteamutils/config.py,sha256=vC6k6E2-Y4mD0E0kw6WVgSatCl9K_BtTwrVFhLrhCzs,665
|
7
|
-
aiteamutils/database.py,sha256=
|
8
|
-
aiteamutils/dependencies.py,sha256=
|
7
|
+
aiteamutils/database.py,sha256=iaqfPTSH3DqauLvwzFQO5XBBsKsL3KrYnu2AuWaTqgQ,32147
|
8
|
+
aiteamutils/dependencies.py,sha256=6YAR2QmGOeL5t9EyWLJHypUopX_CsLDDmmX_3YOSGQQ,5704
|
9
9
|
aiteamutils/enums.py,sha256=ipZi6k_QD5-3QV7Yzv7bnL0MjDz-vqfO9I5L77biMKs,632
|
10
10
|
aiteamutils/exceptions.py,sha256=YV-ISya4wQlHk4twvGo16I5r8h22-tXpn9wa-b3WwDM,15231
|
11
11
|
aiteamutils/security.py,sha256=AZszaTxVEGi1jU1sX3QXHGgshp1lVvd0xXvZejXvs_w,12643
|
12
12
|
aiteamutils/validators.py,sha256=3N245cZFjgwtW_KzjESkizx5BBUDaJLbbxfNO4WOFZ0,7764
|
13
|
-
aiteamutils/version.py,sha256=
|
14
|
-
aiteamutils-0.2.
|
15
|
-
aiteamutils-0.2.
|
16
|
-
aiteamutils-0.2.
|
13
|
+
aiteamutils/version.py,sha256=QQqBt1NyP0N-ZlMtrZ_b51lA-ybgTt3nzbsfJxQS0U8,44
|
14
|
+
aiteamutils-0.2.14.dist-info/METADATA,sha256=On-N3O2UklbAe7Qr8eGvkw831AyfE7iU11JPdJUa3-A,1718
|
15
|
+
aiteamutils-0.2.14.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
|
16
|
+
aiteamutils-0.2.14.dist-info/RECORD,,
|
File without changes
|