保存excel数据到mysql，增加try except防止异常中断，并记录log日志，把异常的文件复制一份到其他文件夹

tuntun · tuntun · commit 5cec24bb5166 · 2015-08-19T09:57:43.000+08:00
diff --git a/save_excel_to_mysql.py b/save_excel_to_mysql.py
@@ -3,6 +3,7 @@
 import MySQLdb
 import os
 import logging
+import shutil
 import sys
 
 
@@ -12,17 +13,21 @@ def main(path):
     """
     files = os.listdir(path)
     for file in files:
-        save_file(path + '/' + file)
+        save_file(path + '/' + file, file)
         print(file)
-        logging.info(file)
 
 
-def save_file(file):
+def save_file(file, filename):
     """
     打开excel文件
     """
-    book = xlrd.open_workbook(file)
-    sheet = book.sheet_by_index(0)
+    try:
+        book = xlrd.open_workbook(file)
+        sheet = book.sheet_by_index(0)
+    except Exception as e:
+        logging.info('错误：{0} 文件：{1}'.format(e, file))
+        shutil.copy2(file, './error' + filename)
+        return False
     row_nums = sheet.nrows
     col_nums = sheet.ncols
 
