Python中为subprocess.call函数参数添加条件判断的方法

在Python中调用bash脚本my_file.sh时，希望仅当params字典中存在指定键（如lang）且对应值不为None时，才向脚本传递该参数。直接在subprocess.call的参数列表中嵌入if语句会触发语法错误：

原错误代码：

import subprocess

def rest_api():
    params = {
        'query': 'indepedence day',
        'formats': '["NEWSPAPER"]',
    }
    subprocess.call(['bash', 
                     'my_file.sh',
                     f'QUERY={params.get("query")}',
                     f'DOC_TYPE={params.get("formats")}',
                     if params.get("lang"): f'LANGUAGE={params.get("lang")}', # 语法错误
                    ])

if __name__ == '__main__':
    rest_api()

报错信息：

Traceback (most recent call last):
  File "nationalbiblioteket_logs.py", line 13, in <module>
    from url_scraping import *
  File "/home/xenial/WS_Farid/DARIAH-FI/url_scraping.py", line 17, in <module>
    from utils import *
  File "/home/xenial/WS_Farid/DARIAH-FI/utils.py", line 53
    if params.get("lang"): f'LANGUAGE={params.get("lang")}',
     ^
SyntaxError: invalid syntax

方法1：分步构建命令列表（直观易维护） #

先初始化基础命令，再根据条件逐步添加参数，逻辑清晰，适合参数较多的场景：

import subprocess

def rest_api():
    params = {
        'query': 'indepedence day',
        'formats': '["NEWSPAPER"]',
    }
    # 初始化基础命令
    cmd = ['bash', 'my_file.sh']
    # 添加必传参数
    cmd.append(f'QUERY={params.get("query")}')
    cmd.append(f'DOC_TYPE={params.get("formats")}')
    # 条件添加可选参数
    lang_val = params.get("lang")
    if lang_val is not None:
        cmd.append(f'LANGUAGE={lang_val}')
    # 执行命令
    subprocess.call(cmd)

if __name__ == '__main__':
    rest_api()

方法2：列表拼接+条件表达式（简洁紧凑） #

利用条件表达式生成可选参数的列表，再与基础命令列表拼接，适合参数较少的场景：

import subprocess

def rest_api():
    params = {
        'query': 'indepedence day',
        'formats': '["NEWSPAPER"]',
    }
    cmd = [
        'bash', 
        'my_file.sh',
        f'QUERY={params.get("query")}',
        f'DOC_TYPE={params.get("formats")}',
    ] + ([f'LANGUAGE={params.get("lang")}'] if params.get("lang") is not None else [])
    subprocess.call(cmd)

if __name__ == '__main__':
    rest_api()

方法3：字典映射批量处理（扩展性强） #

通过字典定义bash变量与params键的映射关系，批量遍历添加参数，后续新增参数只需修改映射字典：

import subprocess

def rest_api():
    params = {
        'query': 'indepedence day',
        'formats': '["NEWSPAPER"]',
    }
    # 定义bash变量名与params键的映射
    param_mapping = {
        'QUERY': 'query',
        'DOC_TYPE': 'formats',
        'LANGUAGE': 'lang'
    }
    cmd = ['bash', 'my_file.sh']
    # 遍历映射，仅当参数存在且不为None时添加
    for bash_var, param_key in param_mapping.items():
        val = params.get(param_key)
        if val is not None:
            cmd.append(f'{bash_var}={val}')
    subprocess.call(cmd)

if __name__ == '__main__':
    rest_api()

Python的列表字面量中不能直接嵌入if语句，因为if是语句而非表达式，无法作为列表的元素存在。必须通过上述方法，将条件判断转化为合法的表达式或分步操作，实现条件添加列表元素。

内容的提问来源于stack exchange，提问作者farid