python安全调用linux命令

使用python调用linux命令时常使用os.system(cmd)的时候，运行成功时返回0，不成功时返回1。

但它本身不会在shell代码报错时停止运行，代码不够安全，容易增加debug的时间。

1. 可以使用断言assert来判断异常：

import os
assert not os.system('echo dir')
assert not os.system("gdalwarp -t_srs EPSG:4326 dbz.tiff dbz_new.tiff")

---------------------------------------------------------------------------
AssertionError                            Traceback (most recent call last)
<ipython-input-18-fe9075e7efc0> in <module>
      1 assert not os.system('echo dir')
----> 2 assert not os.system("gdalwarp -t_srs EPSG:4326 dbz.tiff dbz_new.tiff")

AssertionError: 

可以看到，代码运行会中断，但是不显示具体的报错信息。

2. 使用subprocess

subprocess.run()

Run the command described by args. Wait for command to complete, then return a CompletedProcess instance.

If capture_output is true, stdout and stderr will be captured.

Changed Windows shell search order for shell=True. The current directory and %PATH% are replaced with %COMSPEC% and %SystemRoot%\System32\cmd.exe.

import subprocess as sp
sp.run("gdalwarp -t_srs EPSG:4326 dbz.tiff dbz_new.tiff",shell=True,capture_output=True)

可以看到，返回了CompletedProcess实例，里面包含了stdout和stderr信息，其中returncode最重要，0是正常，1是错误，可以看到了输出的stderr信息。

那对于执行linux命令，尽量封装一个函数，这样代码安全性会更高，也不用去找错误了。

def run_cmd(cmd:str):
    import subprocess as sp
    cp = sp.run(cmd, shell=True, capture_output=True)
    if cp.returncode != 0:
        msg_err = f"""when running the cmd {cmd}: \n {cp.stderr}"""
        raise Exception(msg_err)

调用run_cmd，可以直接看到报错信息了，

报错原因分析： 

安装的GDAL包中包含了指定投影类型的信息，在proj.db中存储了常用的投影系统的参数，因此指定EPSG号就能从proj.db中取到投影信息。
我的情况是在anaconda的虚拟环境中同时安装了GDAL和其他地理空间库，此时同一环境路径下可能会产生多个proj.db文件，访问它的时候程序会无法判断访问哪个。

因此解决方案是：查找gdal安装的那个proj.db在哪里，在程序中指定环境变量。

os.environ['PROJ_LIB'] = r'D:\Anaconda\Lib\site-packages\osgeo\data\proj'

参考资料：

python代码小知识之尽量不使用os.system（） - 简书

python裁剪tif数据，It comes from another PROJ installation._kelly1250230225的博客-CSDN博客