Python数据库sqlite3

时间:2017-01-19 10:14:27

标签: python sqlite

我的代码如下:

def get_code(hex_pattern, database='./AndroidLockScreenRainbow.sqlite'):
    try:
        if os.path.exists(database):
            with lite.connect(database) as db:
                with db.cursor() as c:
                    c.execute("SELECT * FROM RainbowTable")
                    rows = c.fetchall()
                    for row in rows:
                        if row[0] == hex_pattern:
                            return row[1]
        else:
            raise lite.OperationalError("Database file not exists")
    except lite.OperationalError:
        print('Given SQL table not found!')

当代码到达db.cursor()为c:的行时,程序会出现以下错误

with db.cursor() as c:
AttributeError: __exit__

我错了什么?

1 个答案:

答案 0 :(得分:1)

在这种情况下,传递给with语句(db.cursor())的表达式应该返回一个上下文管理器对象。上下文管理器对象必须同时具有__enter____exit__方法(在引擎盖下,with语句使用这些方法来确保正确清理对象)。

sqlite3游标对象没有实现这些方法,所以它不是一个有效的上下文管理器方法,因此你得到的错误信息。

您可以在游标周围编写自己的上下文管理器。您可以自己编写一个,但在大多数情况下这不是必需的,只需将db.cursor()直接分配给db

c = db.cursor()