-
Notifications
You must be signed in to change notification settings - Fork 0
/
xls2csv.py
55 lines (45 loc) · 1.63 KB
/
xls2csv.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
# -*- coding: utf-8 -*-
# Usage:
# python xls2csv.py -in data/input.xlsx -out output/
# python xls2csv.py -in data/input.xlsx -out output/ -sheet "Sheet 1"
import argparse
import csv
import os
import sys
import xlrd
# input
parser = argparse.ArgumentParser()
parser.add_argument('-in', dest="INPUT_FILE", default="path/to/file.xlsx", help="Path to input excel file")
parser.add_argument('-sheet', dest="SHEET", default="", help="Sheet name; leave blank to export all")
parser.add_argument('-out', dest="OUTPUT_DIR", default="path/to/csv/dir/", help="Path to output directory")
# init input
args = parser.parse_args()
def excelToCSV(infile, outdir, sheet=""):
workbook = xlrd.open_workbook(infile)
worksheets = []
# ensure output dir exists
if not os.path.exists(outdir):
os.makedirs(outdir)
# only get one sheet
if len(sheet):
worksheet = workbook.sheet_by_name(sheet)
worksheets.append({
'name': sheet,
'data': worksheet
})
# get all sheets
else:
for worksheet_name in workbook.sheet_names():
worksheet = workbook.sheet_by_name(worksheet_name)
worksheets.append({
'name': worksheet_name,
'data': worksheet
})
# output sheets as csv
for ws in worksheets:
filename = outdir + ws['name'] + '.csv'
with open(filename, 'wb') as f:
wr = csv.writer(f)
for rownum in xrange(ws['data'].nrows):
wr.writerow([unicode(entry).encode("utf-8") for entry in ws['data'].row_values(rownum)])
excelToCSV(args.INPUT_FILE, args.OUTPUT_DIR, args.SHEET)