master 브랜치로 백업

.dockerignore

@@ -4,5 +4,6 @@
 info.log
 db.sqlite3
 .github
-*/migrations
 Dockerfile
+logs
+nginx.conf

.gitignore

@@ -14,3 +14,5 @@ db.sqlite3
 *.dir
 # 로그파일 무시
 *.log
+/logs
+nginx.conf

Dockerfile

@@ -1,8 +1,8 @@
 FROM python:3.12
 
 WORKDIR /code
+RUN apt-get update && apt-get install -y libgl1
 COPY ./requirements.txt /code/
 RUN pip install -r requirements.txt
-RUN apt-get update && apt-get install -y libgl1
 COPY . /code/
 EXPOSE 8000

authenticate/admin.py

@@ -1,3 +1,5 @@
 from django.contrib import admin
 
 # Register your models here.
+from .models import OIDC
+admin.site.register(OIDC)

authenticate/authentications.py

@@ -1,6 +1,7 @@
 import requests
 from rest_framework.authentication import BaseAuthentication
 from rest_framework.exceptions import AuthenticationFailed
+from rest_framework_simplejwt.tokens import AccessToken
 from usr.models import User
 from config.settings import APP_LOGGER
 
@@ -23,17 +24,21 @@ def authenticate(self, request):
         if prefix != 'Bearer':
             raise AuthenticationFailed('Invalid Bearer Prefix')
 
-        # 액세스 토큰 검증을 시도합니다.
-        payload = self.validate_kakao_access_token(access_token)
+
         # 사용자 정보를 받아옵니다.
         try:
-            sub = payload['id']
+            # 액세스 토큰 검증을 시도합니다.
+            payload = AccessToken(access_token)
+            sub = payload['sub']
             user = User.objects.get(sub=sub)
             logger.info(f'username: {user.username}(sub: {sub}) User attempting to access backend Api')
             return user, access_token
         except User.DoesNotExist: # 사용자 정보가 없을 경우
             logger.info('New User attempting to access backend Api')
             return None
+        except Exception as e:
+            logger.info(f'Authentication Failed: {e} - Exception code (a40)')
+            raise AuthenticationFailed(f'Authentication Failed - {e}')
 
     def validate_kakao_access_token(self, access_token):
         end_point = 'https://kapi.kakao.com/v1/user/access_token_info' # 유효성 검증 url

authenticate/tests.py

@@ -24,8 +24,8 @@ def test_login_callback(self):
         """
         redirect_uri = 'http://localhost:8000/auth/login/'
         response = self.client.post(f'/auth/get_token/?code={self.AUTH_CODE}&redirect_uri={redirect_uri}')
-        self.assertEqual(response.status_code, 201)
         print(response.json())
+        self.assertEqual(response.status_code, 201)
 
     @unittest.skipIf(SKIP_TEST == 'True', "Skip Login Refresh Test")
     def test_refresh_token(self):
@@ -53,6 +53,7 @@ def test_refresh_token(self):
         response = self.client.post(target_uri, data=data, content_type='application/json')
         self.assertEqual(response.status_code, 400)
 
+    @unittest.skipIf(SKIP_TEST == 'True', "Skip Login Test")
     def test_login(self):
         """
         해당 함수는 flutter sdk로 발급받은 액세스 토큰과 아이디 토큰을 활용하여 로그인 혹은 회원가입 진행이 되는지 확인합니다.

authenticate/urls.py

@@ -1,12 +1,14 @@
 from django.urls import path
-from .views import kakao_callback, KakaoRefreshTokens, LoginRegisterView
+from .views import kakao_callback, KakaoRefreshTokens, LoginRegisterView, CustomTokenRefreshView
+from rest_framework_simplejwt.views import TokenRefreshView
 
 urlpatterns = [
     path('login/', LoginRegisterView.as_view({
         'post': 'create',
     }), name='login_register'),
     path('get_token/', kakao_callback, name='login'), # 로그인 url 매핑
-    path('refresh/', KakaoRefreshTokens.as_view({
-        'post': 'create',
-    }), name='refresh_tokens'),
+    # path('refresh/', KakaoRefreshTokens.as_view({
+    #     'post': 'create',
+    # }), name='refresh_tokens'),
+    path('refresh/', CustomTokenRefreshView.as_view())
 ]

authenticate/views.py

@@ -3,13 +3,27 @@
 from rest_framework.response import Response
 import requests
 from services.kakao_token_service import KakaoTokenService
+from services.kakao_error_handler import KakaoRequestError
+from rest_framework_simplejwt.views import TokenRefreshView
+from services.exception_handler import *
 
 from usr.services import UserService
+from rest_framework_simplejwt.tokens import RefreshToken
 
-from config.settings import KAKAO_REST_API_KEY # 환경변수를 가져옵니다.
+from config.settings import KAKAO_REAL_NATIVE_API_KEY, KAKAO_REST_API_KEY, APP_LOGGER # 환경변수를 가져옵니다.
+import logging
+logger = logging.getLogger(APP_LOGGER)
 
 # Create your views here.
 
+def get_tokens_for_user(user):
+    refresh = RefreshToken.for_user(user)
+
+    return {
+        'refresh': str(refresh),
+        'access': str(refresh.access_token),
+    }
+
 
 def kakao_callback(request):
     """
@@ -23,33 +37,17 @@ def kakao_callback(request):
     if code is None:
         return JsonResponse({"Error": "인가 코드 추출 실패"}, status=status.HTTP_400_BAD_REQUEST)
     # 토큰을 발급받기위한 클래스 선언
-    token_service = KakaoTokenService()
-    # 요청 body
-    data = {
-        'grant_type': 'authorization_code',
-        'client_id': KAKAO_REST_API_KEY,
-        'redirect_uri': redirect_uri,
-        'code': code,
-    }
-    # 토큰 발급을 요청합니다.
-    token_service.get_kakao_token_response(data)
-    if token_service.status_code == 200:
-        id_token = token_service.id_token
-        # 아이디 토큰이 존재하지 않는다면 -> 예외처리
-        if id_token is None:
-            return JsonResponse({"Error": "id 토큰이 존재하지 않습니다.", "ErrorResponse": token_service.response}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
-        # TODO 유저 생성하여 회원가입 처리 or 로그인 처리
-        user_service = UserService(id_token)
-        user, is_new = user_service.get_or_register_user() # 로그인 혹은 회원가입을 처리합니다.
-        # data 딕셔너리 객체를 생성하여 액세스, 리프레시 토큰만 골라서 추출
-        data = dict()
-        data['access_token'] = token_service.access_token  # 액세스 토큰 추가
-        data['token_type'] = token_service.token_type  # token 타입 정보 추가
-        data['refresh_token'] = token_service.refresh_token  # 리프레시 토큰 정보 추가
-        data['is_new'] = is_new # 신규 유저인지 알려주는 플래그 입니다.
-        # return JsonResponse(data, status=201) # post 요청을 보내줬기 때문에 201 create를 보내줍니다.
-        return JsonResponse(token_service.response, status=status.HTTP_201_CREATED) # 모든 정보를 보내줍니다.
-    return JsonResponse({"Error": token_service.response}, status=status.HTTP_400_BAD_REQUEST)
+    token_service = KakaoTokenService(KAKAO_REST_API_KEY)
+    tokens = token_service.get_tokens(code, redirect_uri)
+    user_service = UserService(tokens.id_token)
+    user, is_new = user_service.get_or_register_user()  # 로그인 혹은 회원가입을 처리합니다.
+    data = dict()
+    data['access_token'] = tokens.access_token  # 액세스 토큰 추가
+    data['token_type'] = tokens.token_type  # token 타입 정보 추가
+    data['refresh_token'] = tokens.refresh_token  # 리프레시 토큰 정보 추가
+    data['id_token'] = tokens.id_token
+    data['is_new'] = is_new # 신규 유저인지 알려주는 플래그 입니다.
+    return JsonResponse(data, status=201) # post 요청을 보내줬기 때문에 201 create를 보내줍니다.
 
 class KakaoRefreshTokens(viewsets.ViewSet):
     """
@@ -62,23 +60,18 @@ def create(self, request):
         if refresh_token is None:
             return Response({"Error": "Refresh token is missing"}, status=400)
         # 리프레시 토큰이 있는경우
-        token_service = KakaoTokenService()
-        data = {
-            'grant_type': 'refresh_token',
-            'client_id': KAKAO_REST_API_KEY,
-            'refresh_token': refresh_token,
-        }
-        token_service.get_kakao_token_response(data) # 카카오 토큰을 재발급 받습니다.
-        # 헤더와 정보를 조합하여 정보를 보냅니다.
-        # 올바른 정보가 넘어왔다면
-        if token_service.status_code == 200:
-            # data 딕셔너리 객체를 생성하여 액세스, 리프레시 토큰만 골라서 추출
+        token_service = KakaoTokenService(KAKAO_REAL_NATIVE_API_KEY)
+        try:
+            tokens = token_service.get_new_tokens(refresh_token) # 카카오 토큰을 재발급 받습니다.
             data = dict()
-            data['access_token'] = token_service.access_token  # 액세스 토큰 추가
-            data['token_type'] = token_service.token_type  # token 타입 정보 추가
+            data['access_token'] = tokens.access_token  # 액세스 토큰 추가
+            data['token_type'] = tokens.token_type  # token 타입 정보 추가
+            data['refresh_token'] = tokens.refresh_token # 리프레시 추가 None 가능
             return Response(data, status=status.HTTP_201_CREATED)
-        # 만일 토큰 정보가 잘못되었거나, refresh_token마저 만료 된경우, 혹은 카카오 측 오류인 경우
-        return Response(token_service.response, status=status.HTTP_400_BAD_REQUEST)
+        except KakaoRequestError as e:
+            return Response({"Error": str(e)}, status=status.HTTP_400_BAD_REQUEST)
+        except Exception as e:
+            return Response({"Error": str(e)}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
 
 class LoginRegisterView(viewsets.ViewSet):
     """
@@ -101,6 +94,10 @@ def create(self, request):
         except Exception as e:
             return Response({"Error": str(e)}, status=status.HTTP_400_BAD_REQUEST)
 
+        tokens = get_tokens_for_user(user)
+        accessToken = tokens['access']
+        refreshToken = tokens['refresh']
+
         return Response({
             "message": "login or register success",
             "is_new": is_new,
@@ -110,6 +107,34 @@ def create(self, request):
                 "profile_image_url": user.profile_image_url,
                 "age_range": user.age_range,
                 "gender": user.gender,
+            },
+            "tokens": {
+                "access_token": accessToken,
+                "refresh_token": refreshToken,
             }
         }, status=status.HTTP_201_CREATED)
 
+class CustomTokenRefreshView(TokenRefreshView):
+    def post(self, request, *args, **kwargs):
+        data = request.data.copy()
+        refresh_token = data.pop('refresh_token', None)
+        if refresh_token is None:
+            raise NoRequiredParameterException(
+                'NO_PARAMETER',
+                'refresh_token 키 값이 존재하지 않습니다.'
+            )
+        data['refresh'] = refresh_token
+        serializer = self.get_serializer(data=data)
+        try:
+            serializer.is_valid(raise_exception=True)
+        except Exception as e:
+            raise ExceptionHandler(
+                'TOKEN_VALIDATION_ERROR',
+                e
+            )
+
+        return Response({
+            'access_token': serializer.validated_data['access'],
+            'token_type': 'Bearer',
+            'refresh_token': serializer.validated_data['refresh'],
+        }, status=status.HTTP_200_OK)

config/settings.py

@@ -39,6 +39,9 @@
 KAKAO_REAL_NATIVE_API_KEY = env('KAKAO_REAL_NATIVE_API_KEY') # 카카오 실제 native api 키
 # 아래는 매 실행마다 코드를 변경해줘야하는 테스트 코드를 임의로 차단하기 위한 환경변수 입니다.
 SKIP_TEST = env('SKIP_TEST')
+GEOCODER_API_KEY = env('GEOCODER_API_KEY')
+KAKAO_REAL_JAVASCRIPT_KEY = env('KAKAO_REAL_JAVASCRIPT_KEY')
+REFRESH_TOKEN = env('REFRESH_TOKEN')
 
 
 # Quick-start development settings - unsuitable for production
@@ -192,6 +195,7 @@
     'DEFAULT_AUTHENTICATION_CLASSES': (
         'authenticate.authentications.CustomAuthentication',
     ),
+    'EXCEPTION_HANDLER': 'services.exception_handler.custom_exception_handler'
 }
 
 # 아래는 celery setting을 담당합니다.
@@ -272,6 +276,9 @@
 }
 
 # 아래는 로그 설정입니다.
+LOG_DIR = './logs'
+os.makedirs(LOG_DIR, exist_ok=True)
+
 LOGGING = {
     'version': 1,
     'disable_existing_loggers': False, # 기본 로거 설정 유지
@@ -281,7 +288,7 @@
             'style': '{', # str.format
         },
         'simple': {
-            'format': '{name} {levelname} {asctime} {message}',
+            'format': '[{levelname}] | {asctime} | {message}',
             'style': '{',
         },
         'logstash': {
@@ -291,22 +298,24 @@
     'handlers': { # 로그 핸들러 설정
         'file': {
             'level': 'DEBUG',
-            'class': 'logging.FileHandler',
-            'filename': 'info.log',
-            'formatter': 'verbose',
-            'encoding': 'utf-8'
+            'class': 'logging.handlers.TimedRotatingFileHandler',
+            'filename': os.path.join(LOG_DIR, 'app.log'),
+            'formatter': 'simple',
+            'encoding': 'utf-8',
+            'when': 'midnight', # 자정마다 새 로그파일 생성
+            'backupCount': 7, # 일주일치만 저장
         },
         'logstash': {
             'level': 'INFO',
             'class': 'config.tcp_log_handler.TCPLogstashHandler',
             'host': env('LOGSTASH_HOST'),
-            'port': 5000,
+            'port': 3306,
             'formatter': 'logstash'
         }
     },
     'loggers': { # 로거 설정, 실제 get_logger를 이용하여 로그 설정 가져옴
         'django': { # 실제 배포 환경에서 사용하는 로거
-            'handlers': ['logstash'],
+            'handlers': ['file'],
             'level': 'INFO',
             'propagate': True,
         },
@@ -319,4 +328,7 @@
 }
 
 # 앱 기본 로거 설정
-APP_LOGGER='django'
+APP_LOGGER='django'
+
+# YOLO 모델 디렉터리 설정
+MODEL_DIR = os.path.join(BASE_DIR, "mission", "yolomodels")