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)