123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081 |
- import xlrd,xlwt
- import datetime
- import uuid
- import os
- from django.conf import settings
- def process_cell(table,i,j):
- _temp=table.cell(i,j)
- if _temp.ctype==1:
- return _temp.value
- if _temp.ctype==2:
- return str(int(_temp.value))
- #raise "Error xls!"
- def randompass():
- return str(uuid.uuid4()).replace('-','')[:10]
- def GeneratePasswdToXls(FILE):
- BASE_DIR = os.path.dirname(os.path.dirname(__file__))
- wrxlsname = 'pwdout-%s.xls' % str(uuid.uuid4())
- wrxlspath = os.path.join(settings.MEDIA_ROOT,'xls',wrxlsname )
- xlsname = os.path.join(BASE_DIR, 'uploads', '%s.xls' % str(uuid.uuid4()))
- destination = open(xlsname,'wb')
- for chunk in FILE.chunks():
- destination.write(chunk)
- destination.close()
- data = xlrd.open_workbook(xlsname)
- table = data.sheet_by_index(0)
- head = table.row_values(0)
- assert len(head)==7
- assert head[0].strip()==u'\u5b66\u53f7'
- assert head[1].strip()==u'\u7528\u6237\u540d'
- assert head[2].strip()==u'\u59d3\u540d'
- assert head[3].strip()==u'\u6027\u522b'
- assert head[4].strip()==u'\u5bc6\u7801'
- assert head[5].strip()==u'\u5206\u7ec4\u53f7'
- assert head[6].strip()==u'\u751f\u65e5'
- rwb = xlwt.Workbook(encoding='utf-8')
- rsh = rwb.add_sheet('Sheet1')
- rsh.write(0,0,u'\u5b66\u53f7')
- rsh.write(0,1,u'\u7528\u6237\u540d')
- rsh.write(0,2,u'\u59d3\u540d')
- rsh.write(0,3,u'\u6027\u522b')
- rsh.write(0,4,u'\u5bc6\u7801')
- rsh.write(0,5,u'\u5206\u7ec4\u53f7')
- rsh.write(0,6,u'\u751f\u65e5')
- for i in range(table.nrows-1):
- col = table.row_values(i+1)
- username=col[1]
- sid=str(int(col[0]))
- name=process_cell(table,i+1,2)
- sex=col[3]
- password=randompass()
- cid=str(int(col[5]))
- if table.cell(i+1,6).ctype==3:
- birthday = datetime.date(*xlrd.xldate_as_tuple(table.cell(i+1,6).value,data.datemode)[:3])
- else:
- try:
- birthday = datetime.datetime.strptime(process_cell(table,i+1,6), "%Y-%m-%d").date()
- except:
- try:
- birthday = datetime.datetime.strptime(process_cell(table,i+1,6), "%Y%m%d").date()
- except:
- pass
-
- rsh.write(i+1,0,sid)
- rsh.write(i+1,1,username)
- rsh.write(i+1,2,name)
- rsh.write(i+1,3,sex)
- rsh.write(i+1,4,password)
- rsh.write(i+1,5,cid)
- rsh.write(i+1,6,birthday.strftime('%Y%m%d'))
- rwb.save(wrxlspath)
- return wrxlsname
-
|