| """Client 企业信息预填 API。""" |
|
|
| from __future__ import annotations |
|
|
| from typing import Any |
|
|
| from ninja import Router |
|
|
| from apps.client.schemas import EnterpriseClientPrefillOut, EnterpriseCompanySearchOut |
| from apps.client.services.client_enterprise_prefill_service import ClientEnterprisePrefillService |
| from apps.core.exceptions import ValidationException |
|
|
| router = Router(tags=["客户管理"]) |
|
|
|
|
| def _get_prefill_service() -> ClientEnterprisePrefillService: |
| return ClientEnterprisePrefillService() |
|
|
|
|
| @router.get("/clients/enterprise/search", response=EnterpriseCompanySearchOut) |
| def search_enterprise_companies( |
| request: Any, |
| keyword: str, |
| provider: str | None = None, |
| limit: int = 8, |
| ) -> EnterpriseCompanySearchOut: |
| if limit < 1 or limit > 20: |
| raise ValidationException(message="limit 必须在 1 到 20 之间", code="INVALID_LIMIT") |
| payload = _get_prefill_service().search_companies(keyword=keyword, provider=provider, limit=limit) |
| return EnterpriseCompanySearchOut(**payload) |
|
|
|
|
| @router.get("/clients/enterprise/prefill", response=EnterpriseClientPrefillOut) |
| def get_enterprise_prefill( |
| request: Any, |
| company_id: str, |
| provider: str | None = None, |
| ) -> EnterpriseClientPrefillOut: |
| payload = _get_prefill_service().build_prefill(company_id=company_id, provider=provider) |
| return EnterpriseClientPrefillOut(**payload) |
|
|
