diff options
| -rw-r--r-- | python/read_excel.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/python/read_excel.py b/python/read_excel.py index d07666f4..ce2f5c43 100644 --- a/python/read_excel.py +++ b/python/read_excel.py @@ -5,15 +5,15 @@ import csv rootdir = "/home/share/rmbs/data/finra/"
-dir_list = os.listdir(rootdir)
+# 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:
+ if os.path.splitext(f)[1] == ".xls":
wb = open_workbook(os.path.join(rootdir, d, f))
- with open( os.path.join(rootdir, d, f.split(".")[0] + ".csv"), "w") as fh:
+ with open( os.path.join(rootdir, d, os.path.splitext(f)[0] + ".csv"), "w") as fh:
wbcsv = csv.writer(fh, dialect = csv.excel)
if "PXTABLES" in f:
s = wb.sheets()[3]
|