使用python操作MongoDB的GridFS下载文件

tuntun · tuntun · commit 22dcadaafbc8 · 2015-08-14T18:20:20.000+08:00
diff --git a/testmongo.py b/testmongo.py
@@ -1,24 +1,31 @@
+# -*- coding:utf-8 -*-
+
 from pymongo import MongoClient
-import gridfs
 import time
 import re
 import os
+import gridfs
 import sys
 
-client = MongoClient('mongodb://192.168.1.22:27017/excel')
+client = MongoClient('mongodb://m_user:m_user_027@192.168.5.22:27017/js_send_excel')
 db = client.js_send_excel
 fs = gridfs.GridFS(db)
+
 files = fs.find()
 
 time_start = time.clock()
 print('&#24635;&#25968;&#65306;', files.count())
+count = 0
 
 for ffle in files:
     filename = ffle.filename
+
     m = re.match(r'&#21457;&#36135;&#35746;&#21333;', filename)
 
-    if m and filename.find('.xls') > 0 and not os.path.isfile('./excel/' + filename):
-        with open('./excel/' + filename, 'wb') as f1:
+    if m and filename.find('xls') > 0 and not os.path.isfile('./excel/' + filename):
+        with open('./excel2/' + filename, 'wb') as f1:
             f1.write(ffle.read())
+            count += 1
 
-print('&#25191;&#34892;&#26102;&#38388;{0}&#31186;'.format(time.clock() - time_start))
+print('&#25191;&#34892;&#26102;&#38388;{0}&#31186;'.format(time.clock() - time_start))
+print('&#26412;&#27425;&#26356;&#26032;{0}&#20010;&#25991;&#26723;'.format(count))
