使用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
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博客