feat: 域名根据domain决定

Author: MistEO

.env.sample

@@ -4,8 +4,4 @@ DATABASE_PORT=3306
 DATABASE_USER=""
 DATABASE_PASSWD=""
 
-ICP_BEIAN=""
-ICP_URL="https://beian.miit.gov.cn/"
-ICP_ENTITY=""
-
 STATIC_APP_DIR="/app/static"

src/ICP/__init__.py

@@ -1,16 +1,34 @@
-from datetime import datetime
 from fastapi import APIRouter, Request
 from loguru import logger
 
-from src.config import settings
+from src.database import ICP
 
 router = APIRouter()
 
+AllICP = ICP.select()
+
 
 @router.get("/icp")
-async def query_icp():
+async def query_icp(request: Request):
+    domain = str(request.base_url)
+
+    icp = next(
+        (icp for icp in AllICP if icp.domain in domain),
+        None,
+    )
+
+    if icp is None:
+        logger.error(f"domain not found: {domain}")
+        return {
+            "domain": domain,
+            "icp_beian": "",
+            "icp_url": "",
+            "icp_entity": "",
+        }
+
     return {
-        "icp_beian": settings.icp_beian,
-        "icp_url": settings.icp_url,
-        "icp_entity": settings.icp_entity,
+        "domain": domain,
+        "icp_beian": icp.beian,
+        "icp_url": icp.url,
+        "icp_entity": icp.entity,
     }

src/config/__init__.py

@@ -7,9 +7,7 @@ class Settings(BaseSettings):
     database_port: int
     database_user: str
     database_passwd: str
-    icp_beian: str
-    icp_url: str
-    icp_entity: str
+    
     static_app_dir: str
 
     class Config:

src/database/__init__.py

@@ -76,6 +76,17 @@ class Meta:
         database = db
         table_name = "project"
 
+class ICP(Model):
+    domain = CharField()
+    beian = CharField()
+    entity = CharField()
+    url = CharField()
+
+    class Meta:
+        database = db
+        table_name = "icp"
+
 Anno.create_table()
 Plan.create_table()
-Project.create_table()
+Project.create_table()
+ICP.create_table()