no message

routers/api/yst/yhxx.py

@@ -2,7 +2,7 @@
 from fastapi import APIRouter, Depends, Request
 from fastapi.responses import PlainTextResponse
 from sqlalchemy.orm import Session
-from database import get_db
+from database import get_db, get_share_db
 from utils.sg_auth import *
 from utils.redis_util import *
 from models import *
@@ -63,12 +63,23 @@ def get_yst_user_info(
 
     corp = x_tif_ext["corp"]
     if corp is not None:
+        with get_share_db() as sharedb:
+            row = sharedb.query(ChemicalCompany).filter(ChemicalCompany.credit_code == corp["social_credit_code"]).first()
+            if row is not None:
+                qymc = row.company_name
+                qyjb = row.safe_level
+                qydz = row.formatted_address
+            else:
+                qymc = corp["name"]
+                qyjb = "-"
+                qydz = "-"
+    
         corp = {
-            "level": x_tif_ext["corp"]["level"],
+            "level": corp["level"],
             "qyInfo": {
-                "qymc": x_tif_ext["corp"]["name"],
-                "qyjb": "--",
-                "qydz": "--"
+                "qymc": qymc,
+                "qyjb": qyjb,
+                "qydz": qydz
             }
         }