如何使用子进程 Popen？

Question

我正在尝试使用 Popen 执行命令。

该命令使用一些 PostGIS/Postgresql 实用程序将光栅文件上传到数据库，并在从命令行执行时工作。它使用 unix 风格的管道来链接 2 个命令，如下所示：

"C:\\Program Files\\PostgreSQL\\9.2\\bin\\raster2pgsql.exe" -d -I -C -e -Y -F -t 128x128 "C:\\temp\\SampleDTM\\SampleDTM.tif" test | "C:\\Program Files\\PostgreSQL\\9.2\\bin\\psql.exe" -h localhost -p 5432 -d adr_hazard -U postgres

在 Python 中使用时，我将其设为带有 ' 代码的字符串：

command = '"C:\\Program Files\\PostgreSQL\\9.2\\bin\\raster2pgsql.exe" -d -I -C -e -Y -F -t 128x128 "C:\\temp\\SampleDTM\\SampleDTM.tif" test | "C:\\Program Files\\PostgreSQL\\9.2\\bin\\psql.exe" -h localhost -p 5432 -d adr_hazard -U postgres'

尝试执行它会导致错误：

p = subprocess.Popen(command)

ERROR: Unable to read raster file: test

错误似乎是命令未正确解析（它将错误的参数解释为光栅文件）

我用Popen 错了吗？

Answer 1

您的command 使用管道|。它需要一个外壳：

p = subprocess.Popen(command, shell=True)

据我所知，command 本身看起来还不错。

Answer 2

不必使用shell=True 来通过管道实现此目的。这可以通过管道以编程方式完成，甚至where concern about insecure input is an issue。这里，conn_params 是一个带有 PASSWORD、NAME（数据库名称）、USER 和 HOST 键的字典。

raster2pgsql_ps = subprocess.Popen([
    'raster2pgsql', '-d', '-I', '-C', '-e', '-Y', '-F', '-t', '128x128',
    'C:\\temp\\SampleDTM\\SampleDTM.tif',
    'test'
], stdout=subprocess.PIPE)

# Connection made using conninfo parameters
# http://www.postgresql.org/docs/9.0/static/libpq-connect.html
psql_ps = subprocess.check_output([
    'psql',
    'password={PASSWORD} dbname={NAME} user={USER} host={HOST}'.format(**conn_params),
], stdin=raster2pgsql_ps.stdout)

Answer 3

以下内容在 Windows 上对我有用，同时避免 shell=True

可以利用 Python 的 fstring formatting 来确保命令在 Windows 中有效。

请注意，我使用了shp2pgsql，但它应该与raster2pgsql 的过程非常相似。

shp2pgsql 的参数：srid 是形状文件的坐标系，filename 是要导入的形状文件的路径，tablename 是您要为表格命名的名称.

import os
import subprocess

shp2pgsql_binary = os.path.join(pgsql_dir, "bin", "shp2pgsql")
psql_binary = os.path.join(pgsql_dir, "bin", "psql")

command0 = f'\"{shp2pgsql_binary}\" -s {srid} \"{filename}\" {tablename}'
command1 = f'\"{psql_binary}\" \"dbname={databasename} user={username} password={password} host={hostname}\"'

try:
    shp2pgsql_ps = subprocess.Popen(command0, stdout=subprocess.PIPE)
    psql_ps = subprocess.check_output(command1, stdin=shp2pgsql_ps.stdout)
except:
    sys.stderr.write("An error occurred while importing data into the database, you might want to \
                        check the SQL command below:")
    sys.stderr.write(command)
    raise

要适应raster2pgsql，你只需要修改command0中的字符串，例如-s {srid} 变为 -d -I -C -e -Y -F -t 128x128。 command1 的字符串可以保持不变。

Answer 4

  PIPE = subprocess.PIPE
  pd = subprocess.Popen(['"C:\\Program Files\\PostgreSQL\\9.2\\bin\\raster2pgsql.exe", '-d', '-I', '-C', '-e', '-Y', '-F', '-t', '128x128', "C:\\temp\\SampleDTM\\SampleDTM.tif", 'test'],
    stdout=PIPE, stderr=PIPE)
  stdout, stderr = pd.communicate()

Answer 5

这样使用subprocess.Popen会更好：

proc = subprocess.Popen(['"C:\\Program Files\\PostgreSQL\\9.2\\bin\\raster2pgsql.exe"', '-d', '-I', '-C', '-e', '-Y', '-F', '-t', '128x128', '"C:\\temp\\SampleDTM\\SampleDTM.tif"', 'test', '|', '"C:\\Program Files\\PostgreSQL\\9.2\\bin\\psql.exe"', '-h', 'localhost', '-p', '5432', '-d', 'adr_hazard', '-U', 'postgres'], shell = True, stdout = subprocess.pipe, stderr = subprocess.STDOUT)
proc.wait()
result = proc.stdout.readlines()#if you want to process the result of your command
proc.kill()

B.T.W，最好先格式化路径，使用：

path = os.path.normalpath("C:\\Program Files\\PostgreSQL\\9.2\\bin\\raster2pgsql.exe")

这将避免不同操作系统平台的一些路径问题。

如果您想像在本地 shell 中执行命令一样执行命令，shell = True 很重要。

希望能帮到你。