From 55d38c107fcdbce6ba6c4196fc186f94012b0d60 Mon Sep 17 00:00:00 2001
From: Coldot <41678750+Coldot@users.noreply.github.com>
Date: Sun, 23 Mar 2025 00:29:01 +0900
Subject: [PATCH] =?UTF-8?q?Feat:=20Cognito=20=ED=81=B4=EB=9D=BC=EC=9D=B4?=
 =?UTF-8?q?=EC=96=B8=ED=8A=B8=20=EC=B4=88=EA=B8=B0=ED=99=94=20=EB=B0=A9?=
 =?UTF-8?q?=EC=8B=9D=20=EA=B0=9C=EC=84=A0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

## 작업 내역
- UserService에서 Cognito 클라이언트를 초기화하는 방식을 개선하여, AWS 세션을 사용하도록 수정
- 사용자 풀 ID를 초기화 시점에 환경 변수에서 가져오도록 설정
---
 src/main/users/service/UserService.py | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/src/main/users/service/UserService.py b/src/main/users/service/UserService.py
index a591723..5920ec7 100644
--- a/src/main/users/service/UserService.py
+++ b/src/main/users/service/UserService.py
@@ -15,18 +15,20 @@
 class UserService:
     def __init__(self):
         self.user_repository = UserRepository()
+        self.user_pool_id = os.environ.get('COGNITO_USER_POOL_ID', '')
         if os.environ.get('ENV') == 'local-profile':
-            self.cognito_client = boto3.client(
-                'cognito-idp',
-                region_name=os.environ.get('AWS_REGION', 'ap-northeast-2'),
+            session = boto3.Session(
                 profile_name=os.environ.get('AWS_PROFILE', 'default')
             )
+            self.cognito_client = session.client(
+                'cognito-idp',
+                region_name=os.environ.get('AWS_REGION', 'ap-northeast-2')
+            )
         else:
             self.cognito_client = boto3.client(
                 'cognito-idp',
                 region_name=os.environ.get('AWS_REGION', 'ap-northeast-2')
             )
-            self.user_pool_id = os.environ.get('COGNITO_USER_POOL_ID', '')
     
     def get_wallets(self, user_id: str):
         wallets = self.user_repository.find_wallets_by_user_id(user_id)