55 lines
1.7 KiB
Python
55 lines
1.7 KiB
Python
import os
|
|
import django
|
|
import pandas as pd
|
|
|
|
os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'TimberTrust.settings')
|
|
django.setup()
|
|
|
|
from business_information.models import Company
|
|
|
|
|
|
def import_companies_from_excel(filename):
|
|
# pandas读取文件
|
|
df = pd.read_excel(filename, engine='openpyxl', skiprows=2)
|
|
# 将数据存入数据库
|
|
for index, row in df.iterrows():
|
|
# 创建公司实例并保存到数据库
|
|
company = Company(
|
|
enterprise_id=index,
|
|
name=row[0],
|
|
operation_status=row[1],
|
|
legal_representative=row[2],
|
|
registered_capital=row[3],
|
|
paid_in_capital=row[4],
|
|
establishment_date=row[5],
|
|
approval_date=row[6],
|
|
operation_period=row[7],
|
|
province=row[8],
|
|
city=row[9],
|
|
district=row[10],
|
|
unified_social_credit_code=row[11],
|
|
taxpayer_identification_number=row[12],
|
|
registration_number=row[13],
|
|
organization_code=row[14],
|
|
insured_persons_number=row[15],
|
|
company_type=row[16],
|
|
industry=row[17],
|
|
used_name=row[18],
|
|
registered_address=row[19],
|
|
latest_annual_report_address=row[20],
|
|
website=row[21],
|
|
phone=row[22],
|
|
other_phone=row[23],
|
|
email=row[24],
|
|
other_email=row[25],
|
|
business_scope=row[26]
|
|
)
|
|
company.save()
|
|
|
|
|
|
# 调用函数的示例
|
|
if __name__ == "__main__":
|
|
# 导入tyc企业信息
|
|
file_path = './files/高级搜索导出数据结果-自定义条件-天眼查(W20031986131710824259889).xlsx'
|
|
import_companies_from_excel(file_path)
|