python - 将文件加载到数据库中

Question

我正在尝试将 csv 文件加载到数据库中，但没有成功。问题是，当我直接在 mysql shell 中加载查询时它可以工作，但当我尝试从 python 代码中执行时它不起作用。我使用的代码如下所示:

db = connect_to_database()
cursor = db.cursor()

q = "LOAD DATA LOCAL INFILE '/home/ubuntu/load_file.csv' INTO TABLE test_load FIELDS TERMINATED BY ',' LINES TERMINATED BY '\n'"
db.query(q)
cursor.execute(q)
db.commit()

它抛出的错误说:

File "/usr/local/lib/python2.7/dist-packages/MySQLdb/connections.py", line 281, in query
    _mysql.connection.query(self, query)
_mysql_exceptions.OperationalError: (1148, 'The used command is not allowed with this MySQL version')

当我将字符串复制到 sql shell 中时，它加载成功。我不确定为什么会抛出此错误。

Answer 1

所以我设法解决了这个问题。它所缺少的只是 MySQLDB 连接中的 local_infile 参数:

db=MySQLdb.connect(host=raw_host,user=raw_user,passwd=raw_passwd,db=raw_db, local_infile = 1)

当我添加此参数时，它按预期工作。感谢大家为解决问题所做的努力。