Python搁置模块问题

Question

Python搁置模块是否具有内置的任何保护以确保两个进程不同时写入文件？

Answer 1

搁置模块使用底层数据库包（如dbm，gdbm或bsddb）。

的restrictions pragraph说（我的重点）：

的货架模块不支持并行读/写访问搁置对象。 （多个同时读取访问是安全的。）当一个程序有一个可写入的架子时，其他程序不应该打开它来读取或写入。 Unix文件锁定可以用来解决这个问题，但是这在Unix版本中是不同的，并且需要关于所使用的数据库实现的知识。

结论：它取决于操作系统和底层数据库。为了保持可移植性，不要建立在并发上。

Answer 2

根据最佳答案，搁置多位作者并不安全。我使货架更安全的方法是编写一个封装，用于打开和访问货架元素。包装代码看起来是这样的：有兴趣的

def open(self, mode=READONLY): 
    if mode is READWRITE: 
     lockfilemode = "a" 
     lockmode = LOCK_EX 
     shelve_mode = 'c' 
    else: 
     lockfilemode = "r" 
     lockmode = LOCK_SH 
     shelve_mode = 'r' 
    self.lockfd = open(shelvefile+".lck", lockfilemode) 
    fcntl.flock(self.lockfd.fileno(), lockmode | LOCK_NB) 
    self.shelve = shelve.open(shelvefile, flag=shelve_mode, protocol=pickle.HIGHEST_PROTOCOL)) 
def close(self): 
    self.shelve.close() 
    fcntl.flock(self.lockfd.fileno(), LOCK_UN) 
    lockfd.close() 

Answer 3

我实现Ivo's approach为上下文管理器，任何人：

from contextlib import contextmanager, closing 
from fcntl import flock, LOCK_SH, LOCK_EX, LOCK_UN 
import shelve 

@contextmanager 
def locking(lock_path, lock_mode): 
    with open(lock_path, 'w') as lock: 
     flock(lock.fileno(), lock_mode) # block until lock is acquired 
     try: 
      yield 
     finally: 
      flock(lock.fileno(), LOCK_UN) # release 

class DBManager(object): 
    def __init__(self, db_path): 
     self.db_path = db_path 

    def read(self): 
     with locking("%s.lock" % self.db_path, LOCK_SH): 
      with closing(shelve.open(self.db_path, "c", 2)) as db: 
       return dict(db) 

    def cas(self, old_db, new_db): 
     with locking("%s.lock" % self.db_path, LOCK_EX): 
      with closing(shelve.open(self.db_path, "c", 2)) as db: 
       if old_db != dict(db): 
        return False 
       db.clear() 
       db.update(new_db) 
       return True