如何用python3自动随机生成Excel文件内容

a305634841

首先来看效果图
文件命名

文件内容

代码说明如下

1. import xlrd
   
2. import xlwt
   
3. from my_framework.log import Logger
   
4. import time,os
   
5. import random
   
6. # create a logger instance
   
7. logger = Logger(logger="file_process").getlog()
   
8. class file_process():
   
9.     """封装操作excel的方法"""
   
10. 
    
11.     # 获取某一页sheet对象
    
12.     def create_excel_file(self):
    
13.         da = time.strftime("%Y-%m-%d/%H-%M-%S", time.localtime())
    
14.         self.da_a = str(da.split('/')[0])
    
15.         self.da_b = str(da.split('/')[1])
    
16.         filename = "case_import"+(self.da_a+'-'+self.da_b).replace('-','.')+".xls"
    
17.         excel_path = os.path.dirname(os.path.abspath('.')) + '\config_file\case_import'
    
18.         self.ecl =os.path.join(excel_path,filename)
    
19. 
    
20.         book = xlwt.Workbook(encoding = 'utf-8')
    
21.         self.sheet1 = book.add_sheet('Sheet 1',cell_overwrite_ok = True)
    
22.         self.title = [u'进件号',u'批次号',u'批次名称',u'产品名称',u'贷款机构',u'产品类型',u'姓名',u'身份证',u'手机',u'年龄',u'性别'u'帐号',
    
23.                 u'卡号',u'逾期天数',u'账龄',u'手别'        ,u'最新欠款金额',u'委案金额',u'委案时间',u'退案时间',u'最近一次进入催收日期',
    
24.                  u'额度',u'案件地区',u'币种',u'开户行',u'账单日',u'开卡日期',u'是否分期',u'分期情况',u'本金',u'利息',u'违约金',
    
25.                  u'最低还款',u'委案最低还款额',u'争议金额',u'争议后金额',u'服务费',u'超限费',u'累计还款',u'委前最后付款额',u'委前最后付款日',
    
26.                 u'案件备注',u'家庭地址',u'家庭电话',u'公司名称',u'公司电话',u'公司地址',u'常用地址',u'账单地址',u'邮政编码',u'其他地址',
    
27.                  u'QQ',u'微信',u'支付宝',u'邮箱',u'联系人1姓名',u'联系人1关系',u'联系人1电话',u'联系人2姓名',u'联系人2关系',u'联系人2电话',
    
28.                  u'联系人3姓名',u'联系人3关系',u'联系人3电话']
    
29. 
    
30.         batch_code = ['SZ-GF-2018-06-20','SZ-GF-2018-06-20','SZ-GF-2018-06-20','SZ-GF-2018-06-20','SZ-GF-2018-06-20','SZ-GF-2018-06-20',
    
31.                       'SZ-GF-2018-06-20','SZ-GF-2018-06-20']
    
32.         commit_money = ['255','452','1223','452','52']
    
33.         self.commit_date = [self.da_a]
    
34.         latest_debt_money = ['255','452','1223','452','52']
    
35.         self.limit_date = ['2019-09-01','2019-05-01','2019-08-21','2019-02-01','2019-12-01','2019-09-25','2019-09-19',]
    
36.         self.product_type = ['信贷','消费金融','信用卡']
    
37.         self.borrower_idnumber = ['422325198908151539','620503198108111211','360103198904054718']
    
38.         self.borrower_phone = ['13686821736']
    
39.         self.loan_institution = ['天天测试银行']
    
40. 
    
41. 
    
42.         self.write_data(self.title,0,line=1)#标题内容和格式写入
    
43. 
    
44. 
    
45.         # 数据导入的条数
    
46.         data_num = 6
    
47.         # 进件号写入
    
48.         bill_code = self.create_data_billcode(data_num)#获取5个随机字符
    
49.         j = self.title_local(u'进件号')
    
50.         self.write_data(bill_code, j, line=0)#对第一列进行写入，写入字段存放在lrst中
    
51.         #批次号写入
    
52.         batch_codes = self.create_data(batch_code,data_num)
    
53.         a = self.title_local(u'批次号')
    
54.         self.write_data(batch_codes, a, line=0)
    
55.         #委案金额写入
    
56.         commit_moneys = self.create_data(commit_money, data_num)
    
57.         a = self.title_local(u'委案金额')
    
58.         self.write_data(commit_moneys, a, line=0)
    
59.         #最新欠款金额写入
    
60.         latest_debt_moneys = self.create_data(latest_debt_money, data_num)
    
61.         a = self.title_local(u'最新欠款金额')
    
62.         self.write_data(latest_debt_moneys, a, line=0)
    
63.         #其它固定写入
    
64.         commit_dates, limit_dates, product_types, borrower_idnumbers, borrower_phones, loan_institutions = self.only_create_data(data_num)
    
65.         #委案时间写入
    
66.         aa = self.title_local(u'委案时间')
    
67.         self.write_data(commit_dates, aa, line=0)
    
68.         #退案时间写入
    
69.         bb = self.title_local(u'退案时间')
    
70.         self.write_data(limit_dates, bb, line=0)
    
71.         #产品类型写入
    
72.         cc = self.title_local(u'产品类型')
    
73.         self.write_data(product_types, cc, line=0)
    
74.         #身份证写入
    
75.         dd = self.title_local(u'身份证')
    
76.         self.write_data(borrower_idnumbers, dd, line=0)
    
77.         #手机号写入
    
78.         ee = self.title_local(u'手机')
    
79.         self.write_data(borrower_phones, ee, line=0)
    
80.         #贷款机构写入
    
81.         ff = self.title_local(u'贷款机构')
    
82.         self.write_data(loan_institutions, ff, line=0)
    
83. 
    
84. 
    
85.         #姓名写入
    
86.         i = self.title_local(u'姓名')
    
87.         brrower_name = self.create_data_name(data_num)  # 获取5个随机字符
    
88.         self.write_data(brrower_name, i, line=0)
    
89.         book.save(self.ecl)
    
90.         return filename
    
91.     def write_data(self,lst,num,line=0):
    
92.         """
    
93. 
    
94.         :param lst: 写入的字段内容的列表
    
95.         :param num: num来控制写入的是固定第几行第几列
    
96.         :param line: 判断写入的是一行还是一列
    
97.         :return: none
    
98.         """
    
99.         if line == 0:
    
100.             for i,item in enumerate(lst):
     
101.                 self.sheet1.write(i+1, num, item)
     
102.         else:
     
103.             for i,item in enumerate(lst):
     
104.                 self.sheet1.write(num, i, item)
     
105.     def title_local(self,str):
     
106.         for i, item in erate(self.title):
     
107.             if item == str:
     
108.                 return i
     
109. 
     
110.     #进件号生成的随机算法
     
111.     def create_data_billcode(self,number):
     
112.         lrst=[]
     
113.         la = ['SZ','PH','HW','KL','WE','TY','DH','Z','H','W','L','E','Y','Q','SZK',
     
114.               'PHK','HWK','KLK','WEK','TYK','DHK','SRZ','PRH','HRW','KRL','WRE','TRY',
     
115.               'DRH','SLZ','PLH','HLW','KLL','WLE','TLY','DLH','SZL','PHL','HWL','KLL','WEL','TYL','DHL']
     
116.         for a in range(1,number+1):
     
117.             i = str(random.randint(0, 10000))
     
118.             j = random.choice(la)
     
119.             ha = j + '-' + self.da_a + '-' + i
     
120.             lrst.append(ha)
     
121.         return lrst
     
122.     #姓名生成的随机算法
     
123.     def create_data_name(self,number):
     
124.         lrst = []
     
125.         la = ['张', '周', '吴', '谢', '李', '高', '汤','曹', '陈', '欧阳', '慕容', '黄', '袁', '吕']
     
126.         lb = ['具', '国', '暮', '各', '色', '形', '者','换', '与', '欧', '各', '力', '固', '体',
     
127.                                     '为', '而', '尔', '风', '哦',
     
128.               '形', '黯', '器', '个', '昂', '瘦', '力', '刻', '他']
     
129.         lc = ['深', '和', '落', '省', '彩', '呈', '的','谎', '和', '妔', '给', '彩', '呈', '的',
     
130.                                 '框', '和', '量', '为', '强',
     
131.               '而', '的', '及', '和', '联', '给', '肉', '开', '费']
     
132.         for a in range(1, number+1):
     
133.             i = random.choice(la)
     
134.             j = random.choice(lb)
     
135.             k = random.choice(lc)
     
136.             ha = i + j + k
     
137.             lrst.append(ha)
     
138.         return lrst
     
139. 
     
140.     #其它数据生成算法
     
141.     def create_data(self,la,number):
     
142.         """
     
143. 
     
144.         :param la: 随机数列
     
145.         :param number: 生成几条数据
     
146.         :return: 数据写集合
     
147.         """
     
148.         lrst = []
     
149.         for a in range(1, number+1):
     
150.             i = random.choice(la)
     
151.             j = str(random.randint(0, 1000))
     
152.             ha = i+j
     
153.             lrst.append(ha)
     
154.         return lrst
     
155.     def only_create_data(self,data_num):
     
156. 
     
157.         commit_dates = []
     
158.         limit_dates = []
     
159.         product_types = []
     
160.         borrower_idnumbers = []
     
161.         borrower_phones = []
     
162.         loan_institutions = []
     
163.         latest_debt_moneys = []
     
164.         for b in range(1,data_num+1):
     
165.             aa = random.choice(self.commit_date)
     
166.             bb = random.choice(self.limit_date)
     
167.             cc = random.choice(self.product_type)
     
168.             dd = random.choice(self.borrower_idnumber)
     
169.             ee = random.choice(self.borrower_phone)
     
170.             ff = random.choice(self.loan_institution)
     
171.             commit_dates.append(aa)
     
172.             limit_dates.append(bb)
     
173.             product_types.append(cc)
     
174.             borrower_idnumbers.append(dd)
     
175.             borrower_phones.append(ee)
     
176.             loan_institutions.append(ff)
     
177.         return commit_dates,limit_dates,product_types,borrower_idnumbers,borrower_phones,loan_institutions
     
178. 
     
179.     #将不是今天生成的文件删除
     
180.     def clear_file(self):
     
181. 
     
182.         da = time.strftime("%Y.%m.%d/%H-%M-%S", time.localtime())
     
183.         da_a = str(da.split('/')[0])
     
184.         path = os.path.dirname(os.path.abspath('.')) + '\config_file\case_import'
     
185.         lrst = []
     
186.         for file in os.listdir(path):
     
187.             if da_a not in file:
     
188.                 lrst.append(file)
     
189.         if len(lrst) == 0:
     
190.             logger.info("没有需要删除的历史文件")
     
191.         else:
     
192.             for file in lrst:
     
193.                 os.remove(os.path.join(path,file))
     
194.     #注意返回的了lsrt是包含标题的
     
195.     def get_data(self,module):
     
196.         lsrt = []
     
197.         line = self.title_local(module)
     
198.         ExcelFile = xlrd.open_workbook(self.ecl)
     
199.         sheet = ExcelFile.sheet_by_index(0)
     
200.         lsrt = sheet.col_values(line)#module列内容
     
201.         return lsrt
     
202. 
     
203. 
     
204. 
     
205. 
     
206. 
     
207. 
     
208. 
     
209. 
     
210. 
     
211. 
     
212. 
     
213. 
     
214. 
     
215. 
     
216. 
     
217. 
     
218. 
     
219. 
     
220. if __name__ == '__main__':
     
221.     a=file_process()
     
222.     a.create_excel_file()
     
223.     a.clear_file()
     
224.     a.get_data(u'姓名')
     

复制代码

软件测试学习资料
百度阅读：
接口测试
UI、功能测试