Python：将光标从类传回dbi实例

Question

尝试设置一个类，该类将把数据库游标传递回来，我可以在各种函数中使用它。 下面的代码不起作用

class create_db(object):   
    def __init__(self):
        import psycopg2 as pq
        self.cn = pq.connect('dbname=mydb user=me')
        self.cr = self.cn.cursor()

我想做的是

cur = create_db()
cur.execute('SELECT * FROM table1;')

如何更改我的代码以解决该问题。

Answer 1

根据您的意愿

class create_db(object):

    def __init__(self):
        import psycopg2 as pq
        self.cn = pq.connect('dbname=mydb user=me')
        self.cr = self.cn.cursor()

    def execute(self, query, *args):

        results = self.cr.execute(query, args)

        return results

现在您可以运行类似

cur.execute('SELECT * FROM table1 WHERE column = ?;', (42,))

我记不起我的头顶，但是我想你想要省略; 在SQL语句的末尾。