编写数据录入
This commit is contained in:
parent
184d95568a
commit
dd518d38f6
|
@ -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
|
|
@ -127,4 +127,4 @@ def create_file(db: Session, file: UploadFile, user_id):
|
||||||
db.refresh(user_file)
|
db.refresh(user_file)
|
||||||
return user_file
|
return user_file
|
||||||
else:
|
else:
|
||||||
raise HTTPException(detail='文件已经上传', status_code=403)
|
return has_uploaded
|
||||||
|
|
Loading…
Reference in New Issue