使用MongoDB來(lái)實(shí)現(xiàn)web.py的session存儲(chǔ)

feiyang10086

使用MongoDB來(lái)實(shí)現(xiàn)web.py的session存儲(chǔ)

web.py 是一個(gè)python的web開(kāi)發(fā)框架，既然涉及到web開(kāi)發(fā)，那么不可避免的可能使用到session，而傳統(tǒng)的session存儲(chǔ)基本都是默認(rèn)在磁盤(pán)，當(dāng)然考慮到web前端機(jī)的擴(kuò)展我們經(jīng)常需要重寫(xiě)session handler來(lái)自定義session的存儲(chǔ)。下面就是在web.py框架中使用MongoDB作session存儲(chǔ)的例子了。

1. from web.session import Store
   
2. import time
   
3. 
   
4. class MongoStore(Store):
   
5.     def __init__(self, db, collection_name):
   
6.         self.collection = db[collection_name]
   
7. 
   
8.     def __contains__(self, key):
   
9.         data = self.collection.find_one({'session_id':key})
   
10.         return bool(data)
    
11. 
    
12.     def __getitem__(self, key):
    
13.         now = time.time()
    
14.         s = self.collection.find_one({'session_id':key})
    
15.         if not s:
    
16.             raise KeyError
    
17.         else:
    
18.             s.update({'attime':now})
    
19.             return s
    
20. 
    
21.     def __setitem__(self, key, value):
    
22.         now = time.time()
    
23. 
    
24.         value['attime'] = now
    
25. 
    
26.         s = self.collection.find_one({'session_id':key})
    
27.         if s:
    
28.             value = dict(map(lambda x: (str(x[0]), x[1]), [(k,v) for (k,v) in value.iteritems() if k not in ['_id']]))
    
29.             s.update(**value)
    
30.             self.collection.save(s)
    
31.         else:
    
32.             self.collection.insert(value)
    
33. 
    
34.     def __delitem__(self, key):
    
35.         self.collection.remove({'session_id':key})
    
36. 
    
37.     def cleanup(self, timeout):
    
38.         timeout = timeout/(24.0*60*60) #timedelta takes numdays as arg
    
39.         last_allowed_time = time.time() - timeout
    
40.         self.collection.remove({'attime' : { '$lt' : last_allowed_time}})

復(fù)制代碼

然后在app中替換掉使用磁盤(pán)進(jìn)行存儲(chǔ)的申明，用

session = web.session.Session(app, MongoStore(db, 'sessions'))替換掉：

session = web.session.Session(app, web.session.DiskStore('sessions'))來(lái)源：python.cn

冰釋一片天

謝謝分享