更新mango工具类

zhanghe06 · zhanghe06 · commit 5902fbba6939 · 2015-11-27T10:33:02.000+08:00
diff --git a/tools/mongo.py b/tools/mongo.py
@@ -107,6 +107,7 @@ def insert(self, table_name, data):
             return ids
         except Exception, e:
             logger.error('插入错误：%s' % e)
+            return None
 
     def update(self, table_name, condition, update_data):
         """
@@ -117,7 +118,13 @@ def update(self, table_name, condition, update_data):
         :param update_data:
         :return:
         """
-        return self.db.get_collection(table_name).update_many(condition, update_data)
+        try:
+            result = self.db.get_collection(table_name).update_many(condition, update_data)
+            logger.info('匹配数量：%s；更新数量：%s' % (result.matched_count, result.modified_count))
+            return result.modified_count  # 返回更新数量，仅支持MongoDB 2.6及以上版本
+        except Exception, e:
+            logger.error('更新失败：%s' % e)
+            return None
 
     def remove(self, table_name, condition=None):
         """
@@ -128,10 +135,11 @@ def remove(self, table_name, condition=None):
         """
         result = self.db.get_collection(table_name).remove(condition)
         if result.get('err') is None:
-            logger.info('删除成功，删除行数%s' % result.get('n'))
+            logger.info('删除成功，删除行数%s' % result.get('n', 0))
+            return result.get('n', 0)
         else:
             logger.error('删除失败：%s' % result.get('err'))
-        return result
+            return None
 
     def output_row(self, table_name, condition=None, style=0):
         """
