feat: update audit platform workspace

fastapi_modules/fastapi_leaudit/models/__init__.py

@@ -17,6 +17,7 @@ from fastapi_modules.fastapi_leaudit.models.usageLoginEvent import UsageLoginEve
 from fastapi_modules.fastapi_leaudit.models.govdocRun import GovdocRun
 from fastapi_modules.fastapi_leaudit.models.govdocRuleResult import GovdocRuleResult
 from fastapi_modules.fastapi_leaudit.models.govdocReportArtifact import GovdocReportArtifact
+from fastapi_modules.fastapi_leaudit.models.qichachaCompanyInfo import QichachaCompanyInfo
 
 __all__ = [
     "LeauditDocument",
@@ -36,4 +37,5 @@ __all__ = [
     "GovdocRun",
     "GovdocRuleResult",
     "GovdocReportArtifact",
+    "QichachaCompanyInfo",
 ]

fastapi_modules/fastapi_leaudit/models/qichachaCompanyInfo.py

@@ -0,0 +1,84 @@
+"""企查查企业信息缓存模型。"""
+
+from __future__ import annotations
+
+from datetime import UTC, datetime
+from typing import Any
+
+from sqlalchemy import BigInteger, Index, String, or_, select
+from sqlalchemy.dialects.postgresql import JSONB
+from sqlalchemy.ext.asyncio import AsyncSession
+from sqlalchemy.orm import Mapped, mapped_column
+
+from fastapi_common.fastapi_common_web.models import BaseModel
+
+
+class QichachaCompanyInfo(BaseModel):
+    """企查查企业信息缓存表。"""
+
+    __tablename__ = "qcc_company_info"
+    __table_args__ = (
+        Index("idx_qcc_company_search_key", "search_key", unique=True),
+        Index("idx_qcc_company_credit_code", "credit_code"),
+        Index("idx_qcc_company_name", "company_name"),
+    )
+
+    Id: Mapped[int] = mapped_column("id", BigInteger, primary_key=True, autoincrement=True)
+    searchKey: Mapped[str] = mapped_column("search_key", String(200), nullable=False, comment="查询关键词")
+    creditCode: Mapped[str | None] = mapped_column("credit_code", String(64), comment="统一社会信用代码")
+    companyName: Mapped[str | None] = mapped_column("company_name", String(255), comment="企业名称")
+    enterprise: Mapped[dict[str, Any] | None] = mapped_column("enterprise", JSONB, comment="工商信息")
+    dishonesty: Mapped[dict[str, Any] | None] = mapped_column("dishonesty", JSONB, comment="失信信息")
+
+    @classmethod
+    async def FindByKeyword(cls, session: AsyncSession, Keyword: str) -> "QichachaCompanyInfo | None":
+        """按查询词、信用代码或企业名查询缓存。"""
+        return await session.scalar(
+            select(cls).where(
+                cls.deleted_at.is_(None),
+                or_(
+                    cls.searchKey == Keyword,
+                    cls.creditCode == Keyword,
+                    cls.companyName == Keyword,
+                ),
+            )
+        )
+
+    @classmethod
+    async def Upsert(
+        cls,
+        session: AsyncSession,
+        SearchKey: str,
+        CreditCode: str | None,
+        CompanyName: str | None,
+        Enterprise: dict[str, Any] | None,
+        Dishonesty: dict[str, Any] | None,
+    ) -> "QichachaCompanyInfo":
+        """按查询关键词写入或更新缓存。"""
+        record = await cls.FindByKeyword(session, SearchKey)
+        if record is None:
+            record = cls(
+                searchKey=SearchKey,
+                creditCode=CreditCode,
+                companyName=CompanyName,
+                enterprise=Enterprise,
+                dishonesty=Dishonesty,
+            )
+            session.add(record)
+        else:
+            record.searchKey = SearchKey
+            record.creditCode = CreditCode
+            record.companyName = CompanyName
+            record.enterprise = Enterprise
+            record.dishonesty = Dishonesty
+            record.updated_at = datetime.now(UTC)
+        await session.flush()
+        return record
+
+    @classmethod
+    def GetAgeDays(cls, record: "QichachaCompanyInfo") -> int:
+        """计算缓存记录距今天数。"""
+        updated_at = record.updated_at
+        if updated_at.tzinfo is None:
+            updated_at = updated_at.replace(tzinfo=UTC)
+        return max((datetime.now(UTC) - updated_at).days, 0)