import uuid from datetime import datetime, timezone from fastapi import Depends from common.database.db_session_manager import DB_SESSION_MNG from common.database.model.models import sessions from common.enums import DBWRType, ErrorType, QuotationStatus, SessionStatus from common.models.gmodel import UserInfo from crud.session_crud import ISessionCRUD, SessionCRUD from router.v1.negotiation.protocol import ListItem, Res_Participate, Res_Reject, Res_SessionList from services.auth_service import AuthService class NegotiationService: """협상 도메인 비즈니스 로직. - 인증(계정 활성 + 저장 토큰 대조)은 AuthService.authenticate 로 위임(재사용). - 목록은 로그인 유저의 supplier_id 로만 조회한다. """ def __init__(self, auth: AuthService = Depends(AuthService), session_crud: ISessionCRUD = Depends(SessionCRUD)): self.auth = auth self.session_crud = session_crud async def list_sessions(self, user_info: UserInfo, access_token: str, status, qt_type, order: str, page: int, page_size: int) -> Res_SessionList: res = Res_SessionList() # 1) 인증 (활성 + 저장된 access 토큰 대조) err_type, info = await self.auth.authenticate(user_info, access_token) if err_type != ErrorType.SUCCESS: res.result.SetResult(err_type) return res supplier_id = uuid.UUID(info.supplier_id) offset = (page - 1) * page_size # 2) 목록 조회 (NEGOTIATION Read 세션, sessions ⨝ items) err_type, rows = await DB_SESSION_MNG.execute_lambda( sessions.DBType(), DBWRType.DB_READ.value, lambda s: self.session_crud.list_by_supplier(s, supplier_id, status, qt_type, order, offset, page_size), ) if err_type != ErrorType.SUCCESS: res.result.SetResult(err_type) return res # 3) 총개수 (페이지네이션용) err_type, total = await DB_SESSION_MNG.execute_lambda( sessions.DBType(), DBWRType.DB_READ.value, lambda s: self.session_crud.count_by_supplier(s, supplier_id, status, qt_type), ) if err_type != ErrorType.SUCCESS: res.result.SetResult(err_type) return res res.items = [ ListItem( session_id=str(r[0]), session_status=r[1], qt_type=r[2], qt_number=r[3], qt_end_time=r[4].isoformat(timespec="seconds") if r[4] else "", item_code=r[5] or "", item_name=r[6] or "", model_name=r[7] or "", maker_name=r[8] or "", ) for r in rows ] res.total = total res.page = page res.page_size = page_size return res async def _load_actionable_session(self, user_info: UserInfo, access_token: str, session_id_str: str, blocked_statuses: tuple): """참여/거부 공통 전처리: 인증 → 세션/견적 로드 → 소유·상태·견적마감·마감시간 검증. 성공 시 (SUCCESS, sess, quote), 실패 시 (err_type, None, None) 을 반환한다. blocked_statuses 에 해당하는 세션 상태면 NEGO_NOT_PARTICIPABLE 로 막는다. """ # 1) 인증 (활성 + 저장된 access 토큰 대조) err_type, info = await self.auth.authenticate(user_info, access_token) if err_type != ErrorType.SUCCESS: return err_type, None, None try: session_id = uuid.UUID(session_id_str) except (ValueError, TypeError): return ErrorType.NEGO_NOT_FOUND, None, None # 2) 세션 조회 err_type, sess = await DB_SESSION_MNG.execute_lambda( sessions.DBType(), DBWRType.DB_READ.value, lambda s: self.session_crud.get_session_by_id(s, session_id), ) if err_type != ErrorType.SUCCESS or sess is None: return ErrorType.NEGO_NOT_FOUND, None, None # 3) 소유 검증 (세션 공급사 == 접속 유저 공급사) if str(sess.supplier_id) != info.supplier_id: return ErrorType.NEGO_FORBIDDEN, None, None # 4) 세션 상태 검증 (호출부가 지정한 불가 상태) if sess.status in blocked_statuses: return ErrorType.NEGO_NOT_PARTICIPABLE, None, None # 5) 견적 조회 + 마감 상태 err_type, quote = await DB_SESSION_MNG.execute_lambda( sessions.DBType(), DBWRType.DB_READ.value, lambda s: self.session_crud.get_quotation_by_id(s, sess.quotation_id), ) if err_type != ErrorType.SUCCESS or quote is None: return ErrorType.NEGO_NOT_FOUND, None, None # 협상완료 세션은 결과 열람용 재진입(무변경)이므로 견적마감·마감시간 검증을 건너뛴다. # reject 는 blocked_statuses 로 DONE 을 이미 막으므로 이 분기는 participate 에만 닿는다. if sess.status == SessionStatus.DONE.value: return ErrorType.SUCCESS, sess, quote if quote.status == QuotationStatus.CLOSED.value: return ErrorType.NEGO_QUOTATION_CLOSED, None, None # 6) 마감 시간 초과 (견적 end_time < 현재). 협상생성(1)일 때만 session→미참여로 정리. end = quote.end_time if end is not None and end.tzinfo is None: end = end.replace(tzinfo=timezone.utc) if end is not None and end < datetime.now(timezone.utc): if sess.status == SessionStatus.CREATED.value: await DB_SESSION_MNG.execute_lambda_run( [sessions.DBType()], [lambda s: self.session_crud.update_session_status(s, sess.session_id, SessionStatus.NOT_PARTICIPATED.value)], ) return ErrorType.NEGO_DEADLINE_PASSED, None, None return ErrorType.SUCCESS, sess, quote async def participate(self, user_info: UserInfo, access_token: str, session_id_str: str) -> Res_Participate: res = Res_Participate() # 미참여/협상거부 상태는 참여 불가 err_type, sess, _ = await self._load_actionable_session( user_info, access_token, session_id_str, (SessionStatus.NOT_PARTICIPATED.value, SessionStatus.REJECTED.value), ) if err_type != ErrorType.SUCCESS: res.result.SetResult(err_type) return res # 참여 성공 — 협상생성(1)일 때만 상태 전이(협상중/완료는 무변경 진입) if sess.status == SessionStatus.CREATED.value: err_type = await DB_SESSION_MNG.execute_lambda_run( [sessions.DBType()], [ lambda s: self.session_crud.update_session_status(s, sess.session_id, SessionStatus.IN_PROGRESS.value), lambda s: self.session_crud.update_quotation_status(s, sess.quotation_id, QuotationStatus.IN_PROGRESS.value), ], ) if err_type != ErrorType.SUCCESS: res.result.SetResult(err_type) return res res.session_id = str(sess.session_id) return res async def reject(self, user_info: UserInfo, access_token: str, session_id_str: str, reject_reason: str) -> Res_Reject: res = Res_Reject() # 거부 사유 필수 reason = (reject_reason or "").strip()[:255] if not reason: res.result.SetResult(ErrorType.INVALID_REQUEST_DATA) return res # 협상완료/미참여/협상거부 상태는 거부 불가 (참여와 공통 검증 재사용) err_type, sess, _ = await self._load_actionable_session( user_info, access_token, session_id_str, (SessionStatus.DONE.value, SessionStatus.NOT_PARTICIPATED.value, SessionStatus.REJECTED.value), ) if err_type != ErrorType.SUCCESS: res.result.SetResult(err_type) return res # 거부 처리 — 세션을 협상거부로 전이하고 사유 저장 err_type = await DB_SESSION_MNG.execute_lambda_run( [sessions.DBType()], [lambda s: self.session_crud.update_session_reject(s, sess.session_id, SessionStatus.REJECTED.value, reason)], ) if err_type != ErrorType.SUCCESS: res.result.SetResult(err_type) return res res.session_id = str(sess.session_id) return res