"""회사 프로필(브랜드) DB 조회 — 자동 온보딩 고객사의 {company_name} 을 DB 에서 채운다. 스크립트의 {company_name} 은 지금까지 tenant.yaml resources.company_name(_base 기본값)에서 왔다. 자동 온보딩 고객사(전용 yaml 없음)는 전부 _base 브랜드로 나가므로, 실제 회사명(company.companies.name)을 company_id 로 조회해 엔진 조립 시 덮어쓴다. 회사 정보라 튜닝 오버레이와는 별개 관심사다. 스키마 소유권: company 스키마는 backend/negodata 소유 — read-only. 경량 table()/column() 구성. """ import uuid from abc import ABC, abstractmethod from typing import Optional, Tuple from sqlalchemy import column, select, table from sqlalchemy.ext.asyncio import AsyncSession from common.database.db_session_manager import DB_SESSION_MNG from common.enums import DBType, DBWRType, ErrorType from common.logger import LOG _COMPANIES = table("companies", column("company_id"), column("name"), column("deleted"), schema="company") class ICompanyProfileRepository(ABC): @abstractmethod async def get_company_name(self, cdb: AsyncSession, company_id) -> Tuple[ErrorType, Optional[str]]: """company.companies.name by company_id. 없으면 None.""" ... class CompanyProfileRepository(ICompanyProfileRepository): async def get_company_name(self, cdb: AsyncSession, company_id) -> Tuple[ErrorType, Optional[str]]: try: query = ( select(_COMPANIES.c.name) .where(_COMPANIES.c.company_id == company_id, _COMPANIES.c.deleted == False) # noqa: E712 .limit(1) ) err_type, rows = await DB_SESSION_MNG.execute(cdb, query, "get_company_name failed.", raise_error=False) if err_type != ErrorType.SUCCESS or not rows or not rows[0]: return err_type, None return ErrorType.SUCCESS, str(rows[0]) except Exception as ex: LOG.e_no_callstack(ex) return ErrorType.DB_RUN_FAILED, None async def resolve_company_name(repo: ICompanyProfileRepository, tenant_key: str) -> Optional[str]: """tenant_key(= company_id UUID)로 회사명 조회. 비-UUID(데모 테넌트)면 조회 안 함(None). 세션/트랜잭션 경계를 여기서 관리(execute_lambda)한다.""" try: cid = uuid.UUID(tenant_key) except (ValueError, TypeError): return None # imarketkorea 등 데모 테넌트명 → 파일 브랜드 유지 async def _q(s): _, name = await repo.get_company_name(s, cid) return name try: return await DB_SESSION_MNG.execute_lambda(DBType.MAIN.value, DBWRType.DB_READ.value, _q) except Exception as ex: LOG.e_no_callstack(f"[company_profile] 회사명 조회 실패 company_id={tenant_key}: {ex}") return None