为了将excel数据自动转换成所需要的erlang数据,听同事说使用python会很方便简单,就自学了两天python,写了一个比较粗糙的python脚本,不过能用,有什么优化的地方请指教
代码如下:
#!/usr/bin/env python # -*- coding: UTF-8 -*- import sys from openpyxl.reader.excel import load_workbook import os import os.path def gen_data(filename): wb = load_workbook('dataxlsx/' + filename + '.xlsx') # 加载文件所有分页 sheetnames = wb.get_sheet_names() # 获取所有分页的名字列表 ws = wb.get_sheet_by_name(sheetnames[0]) # 取第一个分页的数据 # print 'ws:',ws # print "Work Sheet Titile:",ws.title # 分页名称 # print "Work Sheet Rows:",ws.max_row # 分页行数 # print "Work Sheet Cols:",ws.max_column # 分页列数 content = [] # 数据内容 id_list = [] # ID列表 # ========================start concat need data================= content.append('%% this file is auto maked!\n') content.append('-module(' + filename + ').\n') content.append('-compile(export_all).\n') for i in range(4,ws.max_row + 1): # 从表格第三行开始读取,由于range函数不包含文件尾,所以为了读到最后一行需+1 for j in range(ws.max_column): if ws[i][j].value == None: content.append(',""') elif j == 0: id_list.append(int(ws[i][j].value)) content.append('get(' + str(ws[i][j].value).strip() + ') ->\n') content.append(' {r_' + filename + ',' + str(ws[i][j].value).strip()) else: content.append(',' + str(ws[i][j].value).strip()) content.append('};\n') content.append('get(_) ->\n') content.append(' not_match.\n') content.append('length() ->\n') content.append(' ' + str(ws.max_row - 1) + '.\n') content.append('id_list() ->\n ' + str(id_list) + '.') # ==============================end=========================== # 写入数据 f = file('./server/' + filename + '.erl','w+') f.writelines(content) print 'create new file:',filename + '.erl' f.close() # 关闭通道 return def start_gen(): # 删除旧的数据 delnames = os.listdir('./server') for delname in delnames: os.remove('./server/' + delname) print 'delete old file:',delname for _,_,filenames in os.walk('./dataxlsx'): # 遍历文件夹 for filename in filenames: # 遍历文件 find = filename.find('.xlsx') # 返回该文件名称长度 # print "find is:",find if filename[0] == '~' or find == -1: # 文件名以'~'开头或者找不到文件名, 如以'.'开头的文件 continue else: split_list = filename.split('.') # 使用'.'分割文件名,获得[文件名,文件格式] # print split_list gen_data(split_list[0]) # 用文件名作为参数调用gen_data start_gen()
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持编程小技巧。
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。