mongodb在python中的语法糖

点击(此处)折叠或打开

1. #!/usr/bin/env python
   
2. # -*- encoding:utf-8 -*-
   
3. 
   
4. from pymongo import Connection
   
5. from config import database
   
6. import functools
   
7. 
   
8. 
   
9. class Mongodb(object):
   
10.     '''数据库操作辅助类'''
    
11.     def __init__(self):
    
12.         self.db = None
    
13. 
    
14.     def conn(self):
    
15.         if not self.db:
    
16.             connection = Connection(database.database['dbhost'])
    
17.             self.db = connection[database.database['dbname']]
    
18.             # from bottle import
    
19.             # connection = Connection(bottle.config['database.dbhost'])
    
20.             # self.db = connection[bottle.config['database.dbname']]
    
21.         return self.db
    
22. 
    
23.     def find_one(self, name):
    
24.         def wrapper(callback):
    
25.             def wrapper(*args, **kwargs):
    
26.                 conn = self.conn()
    
27.                 query = args[0]
    
28.                 if '_id' in query:
    
29.                     from bson.objectid import ObjectId
    
30.                     if not isinstance(query['_id'], ObjectId):
    
31.                         query['_id'] = ObjectId(query['_id'])
    
32.                 return conn[name].find_one(query)
    
33.             return wrapper
    
34.         return wrapper
    
35. 
    
36.     def find(self, name, select=None, dump=False):
    
37.         def wrapper(callback):
    
38.             def wrapper(*args, **kwargs):
    
39.                 conn = self.conn()
    
40.                 query = args[0]
    
41.                 cursor = conn[name].find(query, select)
    
42.                 if 'limit' in kwargs: #limit
    
43.                     #print 'limit %s' % kwargs['limit']
    
44.                     cursor.limit(int(kwargs['limit']))
    
45.                 if 'skip' in kwargs: #skip
    
46.                     #print 'skip %s' % kwargs['skip']
    
47.                     cursor.skip(int(kwargs['skip']))
    
48.                 if 'order' in kwargs:
    
49.                     order = kwargs['order']
    
50.                     if isinstance(order, dict):
    
51.                         for k in kwargs['order']:
    
52.                             cursor.sort(k, int(kwargs['order'][k]))
    
53.                     elif isinstance(order, tuple):
    
54.                       for k, v in p:
    
55.                           cursor.sort(k, int(v))
    
56.                 return [record for record in cursor] if dump or 'dump' in kwargs and kwargs['dump'] else cursor
    
57.             return wrapper
    
58.         return wrapper
    
59. 
    
60.     def save(self, name):
    
61.         def wrapper(callback):
    
62.             def wrapper(*args, **kwargs):
    
63.                 conn = self.conn()
    
64.                 obj = args[0]
    
65.                 if not obj:
    
66.                     raise ValueError("Not to be save null object")
    
67.                 if 'addtime' not in obj:
    
68.                     import datetime
    
69.                     obj['addtime'] = datetime.datetime.now()
    
70.                 if '_id' in obj:
    
71.                     from bson.objectid import ObjectId
    
72.                     if not isinstance(obj['_id'], ObjectId):
    
73.                         obj['_id'] = ObjectId(obj['_id'])
    
74.                 return conn[name].save(obj)
    
75.             return wrapper
    
76.         return wrapper
    
77. 
    
78.     def remove(self, name):
    
79.         def wrapper(callback):
    
80.             def wrapper(*args, **kwargs):
    
81.                 conn = self.conn()
    
82.                 query = args[0]
    
83.                 if not query: return
    
84.                 if '_id' in query:
    
85.                     from bson.objectid import ObjectId
    
86.                     if not isinstance(query['_id'], ObjectId):
    
87.                         query['_id'] = ObjectId(query['_id'])
    
88.                 return conn[name].remove(query,safe=True)['n']
    
89.             return wrapper
    
90.         return wrapper
    
91. 
    
92. # def make_mongodb_wrapper(name):
    
93. # @functools.wraps(getattr(Mongodb, name))
    
94. # def wrapper(*a, **ka):
    
95. # return getattr(Mongodb(), name)(*a, **ka)
    
96. # return wrapper
    
97. 
    
98. mongo = Mongodb()

用例

点击(此处)折叠或打开

1. import mongo as m
2. 
   
3. entity = {}
   
4. entity['name'] = "username"
5. save(entity)
6. 
   
7. @m.find(name='userTable')
   
8. def find(condition, **kwargs):pass
   
9. 
   
10. @m.remove(name='userTable')
    
11. def delete(condition,**kwargs):pass
    
12. 
    
13. @m.save(name='userTable')
    
14. def save(item):pass
    
15. 
    
16. @m.find_one(name='userTable')
    
17. def find_one(condition):pass