【游戏开发】Excel表格批量转换成CSV的小工具

  1 # -*- coding: utf-8 -*-  
  2 
  3 import os,sys,inspect,re
  4 import xdrlib,xlrd
  5 
  6 # 防止中文乱码
  7 reload(sys)
  8 sys.setdefaultencoding("utf-8")
  9 
 10 # 分割符
 11 C_SPACE = ","
 12 # 结束符
 13 C_END = "\n"
 14 # 输入路径(存放xls文件的路径）
 15 IN_PATH = ""
 16 # 输出路径(导出csv文件的路径)
 17 OUT_PATH = ""
 18 
 19 
 20 # 读取配置文件
 21 def read_config():
 22     config_file = open("config.ini","r")
 23     cur_line = config_file.readline().rstrip("\r\n").split(‘,‘)
 24     global IN_PATH 
 25     IN_PATH = cur_line[1]
 26     cur_line = config_file.readline().rstrip("\r\n").split(‘,‘)
 27     global OUT_PATH 
 28     OUT_PATH = cur_line[1]
 29 
 30 # 过滤路径
 31 def cur_file_dir(path):
 32     if os.path.isfile(path):
 33         path = os.path.dirname(path)
 34     print path
 35     return os.path.abspath(path)
 36 
 37 # 搜索指定文件夹下面的文件
 38 def find_file_by_pattern(pattern=‘.*‘, base=".", circle=True):  
 39     # 查找给定文件夹下面所有xls文件
 40     re_file = re.compile(pattern)
 41     # 第一次搜索的时候过滤下路径，递归之后直接搜索base路径即可
 42     if base == ".":
 43         base = cur_file_dir(IN_PATH)
 44     print u"开始搜索文件夹：",base
 45 
 46     # 存储xls文件的列表
 47     final_file_list = []
 48     # 遍历指定路径下的文件
 49     cur_list = os.listdir(base)  
 50     for item in cur_list:
 51         # 忽略一些干扰的文件，如果你还有其他需要忽略的文件，直接在后面继续添加即可
 52         if item == ".svn":
 53             continue
 54         # 拼接路径
 55         full_path = os.path.join(base, item)
 56         # 忽略临时文件
 57         if full_path.startswith("~"):
 58             continue
 59         # 筛选出xlsx\xls文件
 60         if full_path.endswith(".xlsx") or full_path.endswith(".xls"):
 61             print u"输入文件:" + full_path
 62             bfile = os.path.isfile(item)
 63             if os.path.isfile(full_path):
 64                 if re_file.search(full_path):
 65                     final_file_list.append(full_path)  
 66             else:
 67                 final_file_list += find_file_by_pattern(pattern, full_path)
 68 
 69     # 返回文件列表
 70     return final_file_list
 71 
 72 
 73 # 打开excel
 74 def open_excel(file= ‘file.xls‘):
 75     try:
 76         data = xlrd.open_workbook(file)
 77         return data
 78     except Exception,e:
 79         print str(e)
 80 
 81 
 82 #根据索引获取Excel表格中的数据 参数:file：Excel文件路径, colnameindex：表头列名所在行的索引, by_index：表的索引
 83 def excel_table_byindex(file=‘file.xls‘, colnameindex=0, by_index=0):
 84     data = open_excel(file)
 85     table = data.sheets()[by_index]
 86     nrows = table.nrows #行数
 87     ncols = table.ncols #列数
 88     rowlist = []
 89     
 90     ‘‘‘开始读取数据‘‘‘
 91     for rownum in range(colnameindex, nrows):
 92         rowdata = table.row_values(rownum)
 93         if rowdata:
 94             collist = []
 95             for i in range(ncols):
 96                 collist.append(rowdata[i])
 97             rowlist.append(collist)
 98     return rowlist
 99 
100 #保存csv文件
101 def savaToCSV(_file, _list, _path):
102     filename = ""
103     content = ""
104     #生成文件内容
105     for collist in _list:
106         for i in range(len(collist)):
107             v = collist[i]
108             vstr = ""
109             # print k,v
110             if isinstance(v, float) or isinstance(v, int):
111                 vstr = str(int(v))
112             else:
113                 vstr = v
114             if i > 0:
115                 content = content + C_SPACE
116             content = content + vstr
117         content = content + C_END
118 
119     #生成文件后缀
120     fname = os.path.splitext(_file)
121     filename = fname[0] + ".csv"
122 
123     #写文件
124     if len(filename)>0 and len(content)>0:
125         #filename = OUT_PATH + "/" + filename
126         print u"输出文件:" + filename
127         file_object = open(filename, ‘w‘)
128         file_object.write(content)
129         file_object.close()
130 
131 
132 def main():
133 
134     read_config()
135     filelist = find_file_by_pattern()
136     if len(filelist) > 0:
137         path = ""
138         #遍历文件生成csv
139         for file in filelist:
140             datalist = excel_table_byindex(file, 0)
141             if len(datalist)>0:
142                 savaToCSV(file, datalist, path)
143     else:
144         print u"没有找到任何excel文件！"
145 
146 if __name__=="__main__":
147     main()