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")
