shelve -- 用來持久化任意的Python對象
這幾天接觸了Python中的shelve這個module,感覺比pickle用起來更簡單一些,它也是一個用來持久化Python對象的簡單工具。當我們寫程序的時候如果不想用關系數據庫那么重量級的東東去存儲數據,不妨可以試試用shelve。shelf也是用key來訪問的,使用起來和字典類似。shelve其實用anydbm去創建DB并且管理持久化對象的。
創建一個新的shelf
直接使用shelve.open()就可以創建了
import shelves = shelve.open('test_shelf.db')try: s['key1'] = { 'int': 10, 'float':9.5, 'string':'Sample data' }finally: s.close()如果想要再次訪問這個shelf,只需要再次shelve.open()就可以了,然后我們可以像使用字典一樣來使用這個shelf
import shelves = shelve.open('test_shelf.db')try: existing = s['key1']finally: s.close()print existing當我們運行以上兩個py,我們將得到如下輸出:
$ python shelve_create.py$ python shelve_existing.py{'int': 10, 'float': 9.5, 'string': 'Sample data'} dbm這個模塊有個限制,它不支持多個應用同一時間往同一個DB進行寫操作。所以當我們知道我們的應用如果只進行讀操作,我們可以讓shelve通過只讀方式打開DB:
import shelves = shelve.open('test_shelf.db', flag='r')try: existing = s['key1']finally: s.close()print existing當我們的程序試圖去修改一個以只讀方式打開的DB時,將會拋一個訪問錯誤的異常。異常的具體類型取決于anydbm這個模塊在創建DB時所選用的DB。
寫回(Write-back)
由于shelve在默認情況下是不會記錄待持久化對象的任何修改的,所以我們在shelve.open()時候需要修改默認參數,否則對象的修改不會保存。
import shelves = shelve.open('test_shelf.db')try: print s['key1'] s['key1']['new_value'] = 'this was not here before'finally: s.close()s = shelve.open('test_shelf.db', writeback=True)try: print s['key1']finally: s.close()上面這個例子中,由于一開始我們使用了缺省參數shelve.open()了,因此第6行修改的值即使我們s.close()也不會被保存。
執行結果如下:
$ python shelve_create.py$ python shelve_withoutwriteback.py{'int': 10, 'float': 9.5, 'string': 'Sample data'}{'int': 10, 'float': 9.5, 'string': 'Sample data'} 所以當我們試圖讓shelve去自動捕獲對象的變化,我們應該在打開shelf的時候將writeback設置為True。當我們將writeback這個flag設置為True以后,shelf將會將所有從DB中讀取的對象存放到一個內存緩存。當我們close()打開的shelf的時候,緩存中所有的對象會被重新寫入DB。
import shelves = shelve.open('test_shelf.db', writeback=True)try: print s['key1'] s['key1']['new_value'] = 'this was not here before' print s['key1']finally: s.close()s = shelve.open('test_shelf.db', writeback=True)try: print s['key1']finally: s.close()writeback方式有優點也有缺點。優點是減少了我們出錯的概率,并且讓對象的持久化對用戶更加的透明了;但這種方式并不是所有的情況下都需要,首先,使用writeback以后,shelf在open()的時候會增加額外的內存消耗,并且當DB在close()的時候會將緩存中的每一個對象都寫入到DB,這也會帶來額外的等待時間。因為shelve沒有辦法知道緩存中哪些對象修改了,哪些對象沒有修改,因此所有的對象都會被寫入。
$ python shelve_create.py$ python shelve_writeback.py {'int': 10, 'float': 9.5, 'string': 'Sample data'}{'int': 10, 'new_value': 'this was not here before', 'float': 9.5, 'string': 'Sample data'} {'int': 10, 'new_value': 'this was not here before', 'float': 9.5, 'string': 'Sample data'}最后再來個復雜一點的例子:
#!/bin/env pythonimport timeimport datetimeimport md5import shelveLOGIN_TIME_OUT = 60db = shelve.open('user_shelve.db', writeback=True)def newuser(): global db prompt = "login desired: " while True: name = raw_input(prompt) if name in db: prompt = "name taken, try another: " continue elif len(name) == 0: prompt = "name should not be empty, try another: " continue else: break pwd = raw_input("password: ") db[name] = {"password": md5_digest(pwd), "last_login_time": time.time()} #print '-->', dbdef olduser(): global db name = raw_input("login: ") pwd = raw_input("password: ") try: password = db.get(name).get('password') except AttributeError, e: print "/033[1;31;40mUsername '%s' doesn't existed/033[0m" % name return if md5_digest(pwd) == password: login_time = time.time() last_login_time = db.get(name).get('last_login_time') if login_time - last_login_time < LOGIN_TIME_OUT: print "/033[1;31;40mYou already logged in at: <%s>/033[0m" % datetime.datetime.fromtimestamp(last_login_time).isoformat() db[name]['last_login_time'] = login_time print "/033[1;32;40mwelcome back/033[0m", name else: print "/033[1;31;40mlogin incorrect/033[0m"def md5_digest(plain_pass): return md5.new(plain_pass).hexdigest()def showmenu(): #print '>>>', db global db prompt = """(N)ew User Login(E)xisting User Login(Q)uitEnter choice: """ done = False while not done: chosen = False while not chosen: try: choice = raw_input(prompt).strip()[0].lower() except (EOFError, KeyboardInterrupt): choice = "q" print "/nYou picked: [%s]" % choice if choice not in "neq": print "invalid option, try again" else: chosen = True if choice == "q": done = True if choice == "n": newuser() if choice == "e": olduser() db.close()if __name__ == "__main__": showmenu()感謝閱讀本文,希望能幫助到大家,謝謝大家對本站的支持!
新聞熱點
疑難解答
圖片精選