# -*- coding: utf-8 -*-

import xlrd
from pysnippets.strsnippets import strip

from maintenance.models import ExpressCompanyInfo


def pre_express_company(fpath='./utils/kuaidi/快递100快递公司标准编码-20210922162915.xls'):
    workbook = xlrd.open_workbook(fpath)
    # sheet = workbook.sheet_by_name('SMR')
    sheets = workbook.sheets()
    sheet = sheets[0]
    nrows = sheet.nrows
    for idx in range(2, nrows):
        rvals = sheet.row_values(idx)
        print(rvals)

        name = strip(rvals[0])
        if not name:
            continue

        com = strip(rvals[1])
        if not com:
            continue

        ExpressCompanyInfo.objects.update_or_create(name=name, com=com, defaults={'status': False})