更新mongo工具类

zhanghe06 · zhanghe06 · commit 02aa31043e24 · 2015-11-28T19:37:20.000+08:00
diff --git a/test/test_mongo.py b/test/test_mongo.py
@@ -50,7 +50,9 @@ def test():
         print conn.find_one(table_name)
         print conn.remove(table_name)  # &#28165;&#31354;&#35760;&#24405;
         print conn.insert(table_name, test_date)  # &#25554;&#20837;&#35760;&#24405;
+        print conn.distinct(table_name, 'age')  # &#32479;&#35745;&#24180;&#40836;&#33539;&#22260;
         print conn.update(table_name, {'id': 3}, {'age': 24})  # id=3&#30340;&#35760;&#24405;&#24180;&#40836;&#26356;&#26032;&#20026;24
+        print conn.distinct(table_name, 'age')  # &#32479;&#35745;&#24180;&#40836;&#33539;&#22260;
         conn.output_rows(table_name)
         print conn.update(table_name, {}, {'age': 1}, 'inc')  # &#25152;&#26377;&#35760;&#24405;&#24180;&#40836;&#22686;&#21152;1&#23681;
         conn.output_rows(table_name)
@@ -67,7 +69,9 @@ def test():
 None
 0
 [1, 2, 3]
+[20, 22]
 1
+[20, 22, 24]
 **********  &#34920;&#21517;[user]  [1/3]  **********
 city : shanghai
 name : Lily
diff --git a/tools/mongo.py b/tools/mongo.py
@@ -106,7 +106,7 @@ def insert(self, table_name, data):
             ids = self.db.get_collection(table_name).insert(data)
             return ids
         except Exception, e:
-            logger.error('&#25554;&#20837;&#38169;&#35823;&#65306;%s' % e)
+            logger.error('&#25554;&#20837;&#22833;&#36133;&#65306;%s' % e)
             return None
 
     def update(self, table_name, condition, update_data, update_type='set'):
@@ -120,11 +120,11 @@ def update(self, table_name, condition, update_data, update_type='set'):
         :return:
         """
         if update_type not in ['inc', 'set', 'unset', 'push', 'pushAll', 'addToSet', 'pop', 'pull', 'pullAll', 'rename']:
-            logger.error('&#26356;&#26032;&#31867;&#22411;&#38169;&#35823;&#65306;%s' % update_type)
+            logger.error('&#26356;&#26032;&#22833;&#36133;&#65292;&#31867;&#22411;&#38169;&#35823;&#65306;%s' % update_type)
             return None
         try:
             result = self.db.get_collection(table_name).update_many(condition, {'$%s' % update_type: update_data})
-            logger.info('&#21305;&#37197;&#25968;&#37327;&#65306;%s&#65307;&#26356;&#26032;&#25968;&#37327;&#65306;%s' % (result.matched_count, result.modified_count))
+            logger.info('&#26356;&#26032;&#25104;&#21151;&#65292;&#21305;&#37197;&#25968;&#37327;&#65306;%s&#65307;&#26356;&#26032;&#25968;&#37327;&#65306;%s' % (result.matched_count, result.modified_count))
             return result.modified_count  # &#36820;&#22238;&#26356;&#26032;&#25968;&#37327;&#65292;&#20165;&#25903;&#25345;MongoDB 2.6&#21450;&#20197;&#19978;&#29256;&#26412;
         except Exception, e:
             logger.error('&#26356;&#26032;&#22833;&#36133;&#65306;%s' % e)

-Original file line number
+Diff line change
         print conn.find_one(table_name)
         print conn.remove(table_name)  # 清空记录
         print conn.insert(table_name, test_date)  # 插入记录
 +        print conn.distinct(table_name, 'age')  # 统计年龄范围
         print conn.update(table_name, {'id': 3}, {'age': 24})  # id=3的记录年龄更新为24
 +        print conn.distinct(table_name, 'age')  # 统计年龄范围
         conn.output_rows(table_name)
         print conn.update(table_name, {}, {'age': 1}, 'inc')  # 所有记录年龄增加1岁
         conn.output_rows(table_name)
 None
 [1, 2, 3]
 +[20, 22]
 +[20, 22, 24]
 **********  表名[user]  [1/3]  **********
 city : shanghai
 name : Lily