def test_init_contact(self): """ 初始化联系信息 """ import xlrd book = xlrd.open_workbook('d:/a.xls') sheet = book.sheets()[0] max_row = sheet.nrows print 'begin init database' name_dict = {} max_row = 3 for i in range(2, max_row): print 'row', i dept = sheet.row_values(i)[1].encode("utf-8").strip() job = sheet.row_values(i)[2].encode("utf-8").strip() name = sheet.row_values(i)[3].encode("utf-8").strip() name_pinyin = pinyin.get(name) name_shot = pinyin.get_initial(name, '') username = name_pinyin mobile = str(sheet.row_values(i)[4]) telephone = str(sheet.row_values(i)[5]) innerphone = str(sheet.row_values(i)[6]) fax = str(sheet.row_values(i)[7]) #去掉那些float的. mobile = self.remove_dot(mobile) telephone = self.remove_dot(telephone) innerphone = self.remove_dot(innerphone) fax = self.remove_dot(fax) #统计重复的名字拼音 if name_pinyin in name_dict: username = username + str(name_dict[name_pinyin]) name_dict[name_pinyin] += 1 pass else: name_dict[name_pinyin]=1 print name, job, mobile, telephone, innerphone, fax, dept, pinyin.get(name), name_shot, username #创建到数据库 user = User.objects.get(username='******'); qbinfo = QBContactInfo(duty=job, mobile=mobile, telephone=telephone, innerphone=innerphone, fax=fax) qbinfo.user = user qbinfo.save() print len(name_dict) print name_dict print 'end'
def init_base_data(request): import xlrd import pinyin book = xlrd.open_workbook('d:/a.xls') sheet = book.sheets()[0] max_row = sheet.nrows print 'begin init database' name_dict = {} # max_row = 3 for i in range(2, max_row): print 'row', i dept_name = sheet.row_values(i)[1].encode("utf-8").strip() job = sheet.row_values(i)[2].encode("utf-8").strip() name = sheet.row_values(i)[3].encode("utf-8").strip() name_pinyin = pinyin.get(name) name_shot = pinyin.get_initial(name, '') first_name = name[:2] last_name = name[2:] username = name_pinyin mobile = str(sheet.row_values(i)[4]) telephone = str(sheet.row_values(i)[5]) innerphone = str(sheet.row_values(i)[6]) fax = str(sheet.row_values(i)[7]) #去掉那些float的. mobile = remove_dot(mobile) telephone = remove_dot(telephone) innerphone = remove_dot(innerphone) fax = remove_dot(fax) #统计重复的名字拼音 if name_pinyin in name_dict: username = username + str(name_dict[name_pinyin]) name_dict[name_pinyin] += 1 pass else: name_dict[name_pinyin]=1 # print name, job, mobile, telephone, innerphone, fax, dept_name, pinyin.get(name), name_shot, username #创建部门 dept = None try: dept = QBDepartment.objects.get(dept_name=dept_name) except QBDepartment.DoesNotExist: dept = QBDepartment(dept_name=dept_name) dept.save() #创建用户到数据库 user = User.objects.create_user(username, None, username, first_name=first_name, last_name=last_name); user.save() #创建联系信息 qbinfo = QBContactInfo(name=name, duty=job, mobile=mobile, telephone=telephone, innerphone=innerphone, fax=fax, name_pinyin=name_pinyin, name_shot=name_shot) qbinfo.user = user qbinfo.qbdepartment = dept qbinfo.save() return HttpResponse("OK")