blob: d07666f415fe18c68ba95a145d82a4a4667b94a0 (
plain)
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
|
import os
from xlrd import open_workbook
import csv
rootdir = "/home/share/rmbs/data/finra/"
dir_list = os.listdir(rootdir)
dir_list = [d for d in dir_list if "zip" not in d]
for d in dir_list:
file_list = os.listdir(os.path.join(rootdir, d))
for f in file_list:
if "xls" in f:
wb = open_workbook(os.path.join(rootdir, d, f))
with open( os.path.join(rootdir, d, f.split(".")[0] + ".csv"), "w") as fh:
wbcsv = csv.writer(fh, dialect = csv.excel)
if "PXTABLES" in f:
s = wb.sheets()[3]
else:
s = wb.sheets()[0]
for row in range(s.nrows):
this_row = []
for col in range(s.ncols):
val = s.cell_value(row, col)
if isinstance(val, unicode):
val = val.encode('utf8')
if isinstance(val, float):
val = str(val)
this_row.append(val)
wbcsv.writerow(this_row)
|
