用Python创建MySQL游标函数

时间:2015-03-22 21:04:51

标签: python mysql

我正在尝试创建一个初始化MySQL + Python连接的函数。以下是我到目前为止的情况:

import mysql.connector

MySQLConfig = {
  'user': 'user',
  'password': 'password',
  'host': 'localhost',
  'database': 'DB',
  'raise_on_warnings': True,
}

# Initialize MySQL Connection + Cursor
def createCursor():
    cnx = mysql.connector.connect(**MySQLConfig)
    cursor = cnx.cursor()
    return cursor

cursor = createCursor()

query = 'SHOW tables'
cursor.execute(query)

for row in cursor:
    print(row)

如果没有包含在createCursor()函数中,代码将正常运行。一旦我把它放在一个我收到以下错误:

Traceback (most recent call last):
  File "./config.py", line 24, in <module>
    cursor.execute(query)
  File "/Library/Python/2.7/site-packages/mysql/connector/cursor.py", line 473, in execute
    if not self._connection:
ReferenceError: weakly-referenced object no longer exists

关于我可能需要做什么的任何想法?我已经尝试仅返回连接并使用光标在函数外部,这也会导致相同的错误。

1 个答案:

答案 0 :(得分:2)

由于你已经使cnx成为局部变量,所以在函数结束时会收集垃圾。
做这样的事情

cnx = mysql.connector.connect(**MySQLConfig)
# Initialize  Cursor
def createCursor():
    return cnx.cursor()

但这是一个坏主意。我的方法就是这样的

import mysql.connector
class MysqlHelper:

    def __init__(self):
        MySQLConfig = {'user': 'user','password': 'password','host': 'localhost','database': 'DB','raise_on_warnings': True}
        self.cnx = mysql.connector.connect(**MySQLConfig) 
        self.cursor = self.cnx.cursor()

    def execute_query(self,query):
        self.cursor.execute(query)      
        for row in self.cursor:
            print(row)

mysql_helper = MysqlHelper()
mysql_helper.execute_query("SHOW tables")