From dd518d38f68ea7885e614c35164d6f928f03f8a1 Mon Sep 17 00:00:00 2001
From: wcq <wuchunquan@fecr.com.cn>
Date: Wed, 29 Mar 2023 17:03:19 +0800
Subject: [PATCH] =?UTF-8?q?=E7=BC=96=E5=86=99=E6=95=B0=E6=8D=AE=E5=BD=95?=
 =?UTF-8?q?=E5=85=A5?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 DataLoad/CompanyDataLoad.py | 36 ++++++++++++++++++++++++++++++++++++
 Mods/User/Crud.py           |  2 +-
 2 files changed, 37 insertions(+), 1 deletion(-)
 create mode 100644 DataLoad/CompanyDataLoad.py

diff --git a/DataLoad/CompanyDataLoad.py b/DataLoad/CompanyDataLoad.py
new file mode 100644
index 0000000..520ca2e
--- /dev/null
+++ b/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
diff --git a/Mods/User/Crud.py b/Mods/User/Crud.py
index 62df05c..f6d733f 100644
--- a/Mods/User/Crud.py
+++ b/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