编写数据录入
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)
|
||||
return user_file
|
||||
else:
|
||||
raise HTTPException(detail='文件已经上传', status_code=403)
|
||||
return has_uploaded
|
||||
|
|
Loading…
Reference in New Issue