编写数据录入

DataLoad/CompanyDataLoad.py

@@ -0,0 +1,36 @@
+# 数据录入
+from sqlalchemy.orm import Session
+from Mods.BasicInformation.Mods.CompanyIndexMain.Models import CompanyIndexMain
+from Mods.BasicInformation.Mods.CompanyBaseInfo.Models import CompanyBaseInfo
+from Context.common import tianyancha_api, yujingtong_api
+
+
+def get_company_id(db: Session, company_name):
+    company = db.query(CompanyIndexMain).filter_by(company_name=company_name).first()
+    if not company:
+        raise Exception("公司名称未录入数据库")
+    return company.company_id
+
+
+def company_data_load(db: Session, company_name):
+    pass
+
+
+def add_company_base_info(db: Session, company_name, company_id=None):
+    if not company_id:
+        company_id = get_company_id(company_name)
+    item = db.query(CompanyBaseInfo).filter_by(company_id=company_id).first()
+    if not item:
+        res = tianyancha_api.get_company_base_info(company_name)
+        new_item = CompanyBaseInfo()
+        new_item.company_id = company_id
+        new_item.legal_person = res.legalPersonName
+        new_item.reg_capital = res.regCapital
+        # new_item.paid_capital =
+        new_item.estiblish_date = res.estiblishTime
+        new_item.approved_date = res.approvedTime
+        # new_item.license_duration=
+        new_item.reg_capital = res.regLocation
+        new_item.credit_code = res.creditCode
+        new_item.social_staff_num = res.socialStaffNum
+        new_item.business_scope = res.businessScope

Mods/User/Crud.py

@@ -127,4 +127,4 @@ def create_file(db: Session, file: UploadFile, user_id):
         db.refresh(user_file)
         return user_file
     else:
-        raise HTTPException(detail='文件已经上传', status_code=403)
+        return has_uploaded