使用子进程导入SQL转储

Question

我正在尝试通过Python和子进程将.sql转储从磁盘导入到MySQL中。即等效于

mysql -u user -ppassword db < dump.sql

我的Python代码看起来像这样(但我已经尝试了很多替代方法：)：

proc = subprocess.Popen(
    ("mysql -u %s -p%s database"  % (MYSQL_USER, MYSQL_PASSWORD)).split(),
    stdin=subprocess.PIPE,
    stdout=subprocess.PIPE,
    shell=False)
out, err = proc.communicate('source /tmp/dump.sql')

应用程序成功完成，但没有行导入到MySQL。我也尝试过像这样通过管道连接dump.sql：

proc = subprocess.Popen(
    ("mysql -u %s -p%s database < /tmp/dump.sql"  % (MYSQL_USER, MYSQL_PASSWORD)).split(),
    stdin=subprocess.PIPE,
    stdout=subprocess.PIPE,
    shell=False)
out, err = proc.communicate()

如果重要，当我设置shell=True时，我会得到ERROR 1045 (28000): Access denied for user 'root'@'localhost' (using password: NO))

有谁能给我指个方向吗？

Answer 1

您错误地使用了Popen.communicate()。

import subprocess

proc = subprocess.Popen(["mysql", "--user=%s" % USER, "--password=%s" % PASS, "database"],
                        stdin=subprocess.PIPE,
                        stdout=subprocess.PIPE)
out, err = proc.communicate(file("/tmp/dump.sql").read())

Answer 2

如果您从Google访问此页面，请注意，sigi答案将有效，但它会将所有转储文件加载到内存中，如果它太大而无法容纳，它将失败。

我是这样做的：

with open(dump_filename, 'r') as f: 
       command = ['mysql', '-u%s' % db_settings['USER'], '-p%s' % db_settings['PASSWORD'], db_settings['NAME']]
       proc = subprocess.Popen(command, stdin = f)
       stdout, stderr = proc.communicate()

它也做同样的事情，但是内存消耗很小，因为转储直接流到mysql的stdin。