Browse Source

250210-1代码。

baoyubo 3 months ago
parent
commit
22831b2843
1 changed files with 14 additions and 7 deletions
  1. 14 7
      routers/api/companyManagement/__init__.py

+ 14 - 7
routers/api/companyManagement/__init__.py

@@ -175,7 +175,11 @@ async def create_company(
         traceback.print_exc()
         raise HTTPException(status_code=500, detail=str(e))
 
-
+def area_code_get_area_name(db,area_code):
+    query = db.query(GovdataArea).filter_by(area_code=area_code).first()
+    if query:
+        return query.area_name
+    return ''
 @router.put('/company/update')
 async def update_company(
     db: Session = Depends(get_db),
@@ -193,18 +197,21 @@ async def update_company(
             return JSONResponse(status_code=404, content={"code": 404, "msg": "数据不存在"})
         if 'company_name' in body:
             company.company_name = body['company_name']
-        if 'province' in body:
-            company.province = body['province']
+        # if 'province' in body:
+        #     company.province = body['province']
         if 'province_code' in body:
             company.province_code = body['province_code']
-        if 'city' in body:
-            company.city = body['city']
+            company.province = area_code_get_area_name(db,body['province_code'])
+        # if 'city' in body:
+        #     company.city = body['city']
         if 'city_code' in body:
             company.city_code = body['city_code']
-        if 'district' in body:
-            company.district = body['district']
+            company.city = area_code_get_area_name(db,body['city_code'])
+        # if 'district' in body:
+        #     company.district = body['district']
         if 'district_code' in body:
             company.district_code = body['district_code']
+            company.district =area_code_get_area_name(db,body['district_code'])
         if 'company_address' in body:
             company.company_address = body['company_address']
         if 'responsible_person' in body: