"""风评域仓储：风评记录 + 客户画像（画像主键为 customer_id）。"""
from __future__ import annotations

from sqlalchemy import select

from model.fin_customer_profile import FinCustomerProfile
from model.fin_risk_assessment import FinRiskAssessment
from repositories.base import BaseRepository


class RiskAssessmentRepo(BaseRepository):
    model = FinRiskAssessment


class CustomerProfileRepo(BaseRepository):
    """客户画像仓储。注意：主键是 customer_id 而非 id，不适用基类 get(pk)。"""

    model = FinCustomerProfile

    async def get_by_customer(self, customer_id: int) -> FinCustomerProfile | None:
        return await self.db.scalar(
            select(FinCustomerProfile).where(
                FinCustomerProfile.customer_id == customer_id
            )
        )

    async def upsert_risk(
        self, customer_id: int, risk_level: str, risk_score: int
    ) -> FinCustomerProfile:
        """回写风险等级与评分：无画像则初始化，有则更新并递增画像版本号。"""
        profile = await self.get_by_customer(customer_id)
        if profile is None:
            profile = FinCustomerProfile(
                customer_id=customer_id,
                risk_level=risk_level,
                risk_score=risk_score,
                profile_version=1,
            )
            self.db.add(profile)
        else:
            profile.risk_level = risk_level
            profile.risk_score = risk_score
            profile.profile_version = (profile.profile_version or 0) + 1
        await self.db.commit()
        await self.db.refresh(profile)
        return profile