57 lines
2.2 KiB
Python
57 lines
2.2 KiB
Python
"""tkuser API 프록시 클라이언트 — tkeg"""
|
|
import httpx
|
|
import os
|
|
from typing import Optional
|
|
from fastapi import HTTPException, Request
|
|
|
|
TKUSER_API_URL = os.getenv("TKUSER_API_URL", "http://tkuser-api:3000")
|
|
|
|
|
|
def get_token_from_request(request: Request) -> str:
|
|
auth = request.headers.get("Authorization", "")
|
|
if auth.startswith("Bearer "):
|
|
return auth[7:]
|
|
return request.cookies.get("sso_token", "")
|
|
|
|
|
|
def _headers(token: str) -> dict:
|
|
if not token:
|
|
raise HTTPException(status_code=401, detail="인증 토큰이 필요합니다.")
|
|
return {"Authorization": f"Bearer {token}"}
|
|
|
|
|
|
def _map_project(data: dict) -> dict:
|
|
return {
|
|
"id": data.get("project_id"),
|
|
"job_no": data.get("job_no"),
|
|
"project_name": data.get("project_name"),
|
|
"client_name": data.get("client_name"),
|
|
"is_active": data.get("is_active", True),
|
|
"created_at": data.get("created_at"),
|
|
}
|
|
|
|
|
|
async def get_projects(token: str, active_only: bool = True) -> list:
|
|
endpoint = "/api/projects/active" if active_only else "/api/projects"
|
|
async with httpx.AsyncClient(timeout=10.0) as client:
|
|
resp = await client.get(f"{TKUSER_API_URL}{endpoint}", headers=_headers(token))
|
|
if resp.status_code != 200:
|
|
raise HTTPException(status_code=resp.status_code, detail="프로젝트 목록 조회 실패")
|
|
body = resp.json()
|
|
if not body.get("success"):
|
|
raise HTTPException(status_code=500, detail=body.get("error", "알 수 없는 오류"))
|
|
return [_map_project(p) for p in body.get("data", [])]
|
|
|
|
|
|
async def get_project(token: str, project_id: int) -> Optional[dict]:
|
|
async with httpx.AsyncClient(timeout=10.0) as client:
|
|
resp = await client.get(f"{TKUSER_API_URL}/api/projects/{project_id}", headers=_headers(token))
|
|
if resp.status_code == 404:
|
|
return None
|
|
if resp.status_code != 200:
|
|
raise HTTPException(status_code=resp.status_code, detail="프로젝트 조회 실패")
|
|
body = resp.json()
|
|
if not body.get("success"):
|
|
raise HTTPException(status_code=500, detail=body.get("error", "알 수 없는 오류"))
|
|
return _map_project(body.get("data"))
|