如何在Python文件中调用依赖命令行的Python脚本及传参API？

嘿，我来帮你搞定这个问题！你现在碰到的是「想在Python脚本里调用原本只能通过命令行触发的API工具」的常见场景，给你两种靠谱的解决方案，按需选择：

如果不想改动原有的pythonapi.py，直接用Python内置的subprocess模块模拟命令行调用是最直接的办法。这种方式完全复刻你手动敲命令的逻辑，还能避免手动拼接命令字符串的转义坑。

举个具体的代码示例：

import subprocess
import json

# 先构造查询字典，比手动写JSON字符串更不容易出错
query_params = {
    "query": {
        "operator": "all",
        "children": [
            {"field": "alias.ip_address", "operator": "contains", "value": "192.168.1.60"}
        ]
    },
    "scope": "private",
    "size": 10  # 补全你需要的size值
}
# 把字典转成标准JSON字符串
query_json = json.dumps(query_params)

# 构造命令参数列表（subprocess推荐用列表形式，避免shell=True的安全风险）
command = [
    "python",
    "bin/pythonapi.py",
    "-K", "72387bc9236d73-13c1-4r1g-83c8-26329b981147",
    "--samples",
    "-j",
    "-r", query_json
]

# 执行命令并捕获输出
result = subprocess.run(command, capture_output=True, text=True)

# 处理执行结果
if result.returncode == 0:
    print("API调用成功：", result.stdout)
else:
    print("API调用失败，错误信息：", result.stderr)

小提示： 用列表形式传递命令参数，subprocess会自动帮你处理所有转义问题，比直接拼字符串安全得多（尤其是当参数里有特殊字符或者来自外部输入时）。

如果这个API是你自己维护的，或者你有权限修改代码，那把它改造成可导入调用的形式会更优雅——毕竟你之前已经习惯了导入Python文件调用功能。

第一步：修改bin/pythonapi.py的结构 #

原文件大概率是用argparse处理命令行参数的，你只需要把核心逻辑抽成独立函数，保留命令行入口即可：

import argparse
import json  # 如果需要处理JSON的话

# 把核心逻辑抽成独立函数，接收明确的参数
def run_api(api_key, samples=False, json_output=False, query_json=None):
    # 这里放原来main函数里解析参数后的业务逻辑
    # 比如先把query_json转成字典：
    query = json.loads(query_json)
    # 执行API请求、处理结果等操作
    # ...
    return processed_result

# 保留原有的命令行入口，方便继续用命令行调用
def main():
    parser = argparse.ArgumentParser()
    parser.add_argument("-K", dest="api_key", required=True, help="API密钥")
    parser.add_argument("--samples", action="store_true", help="是否返回样本数据")
    parser.add_argument("-j", dest="json_output", action="store_true", help="是否输出JSON格式")
    parser.add_argument("-r", dest="query_json", required=True, help="查询条件JSON字符串")
    args = parser.parse_args()
    # 调用抽出来的核心函数
    result = run_api(**vars(args))
    # 处理输出（比如如果json_output为True就打印JSON）
    # ...

if __name__ == "__main__":
    main()

第二步：在你的脚本里直接导入调用 #

现在你可以像导入普通Python模块一样调用这个API了：

from bin.pythonapi import run_api
import json

# 构造查询参数
query_params = {
    "query": {
        "operator": "all",
        "children": [
            {"field": "alias.ip_address", "operator": "contains", "value": "192.168.1.60"}
        ]
    },
    "scope": "private",
    "size": 10
}
query_json = json.dumps(query_params)

# 直接调用核心函数
api_result = run_api(
    api_key="72387bc9236d73-13c1-4r1g-83c8-26329b981147",
    samples=True,
    json_output=True,
    query_json=query_json
)

# 处理结果
print(api_result)

这种方式的好处是：不需要启动子进程，调用更直接，调试起来更方便，参数传递也更安全，完全符合Python的编码习惯。

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