在python 2.7中使用文件锁定的SQlite3上下文管理器

时间:2017-12-04 11:25:30

标签: python python-2.7 sqlite

这是在Python 2.7下的上下文管理器中重新进入文件锁定的好方法吗?我只是想确保rlock()有效,所以我可以使用多线程应用程序来使用单个数据库文件。

import sqlite3
import threading
import os

class ConnectionHolder:
    def __init__(self, connection):
        self.path = connection
        self.lock = threading.RLock()

    def __enter__(self):
        self.lock.acquire()
        self.connection = sqlite3.connect(self.path)
        self.cursor = self.connection .cursor()
        return self.cursor


    def __exit__(self, exc_class, exc, traceback):
        self.connection .commit()
        self.connection .close()
        self.lock.release()

conn_holder = ConnectionHolder(os.path.join(os.path.dirname(__file__), 'data/db/database.db'))

if __name__ == '__main__':
    with conn_holder as c:
        c.execute("SELECT * FROM 'sample_table'")
        result = c.fetchall()
        print result

1 个答案:

答案 0 :(得分:0)

我终于在我的代码中喜欢我在提交之前运行一个长循环(大约2分钟)的地方。我纠正了这个问题,正如@DB建议的那样,将忙碌超时增加到30秒。问题似乎得到了解决。谢谢你们!