Python函数：使用批处理文件将参数从.txt文件传递给python函数并执行函数

Posted 2023-02-28

我想使用批处理文件将一个或多个参数从文本文件传递到python函数。这可能吗？理想情况下，我想从文本文件中的一行读取，该行将特定的com端口传递给python函数my_function，这些操作可以使用批处理文件完成

我现在可以使用批处理文件调用python脚本，如下所示。另外我还可以调用python函数并使用Python Shell将参数传递给它。我需要能够将不同的值从文本文件传递到我遇到的同一个函数。

任何帮助将不胜感激。

当前批处理文件代码调用python脚本

echo[
@echo. The Step below calls the script which opens COM 12
echo[

"C:\Python\python.exe" "C:\Scripts\open_COM12.py"

当前的python代码传递参数（com端口号）并调用python函数

import ConfigComPort as cw
from ConfigComPort import my_function
my_function('12')

连接成功

文本文件内容

COM_PORTS
12
19
23
22

答案

如果你有一个名为parameters.txt的数据文件

foo
bar
foobar

还有一个功能

def my_function(some_text):
    print("I was called with " + some_text)

然后你可以这样做将文件的每一行传递给函数：

with open('parameters.txt', 'r') as my_file:
    for line in my_file:
        # remove the # and space from the next line to enable output to console:
        # print(line.rstrip())

        my_function(line.rstrip())

请注意，我所有示例中的rstrip()方法都会删除尾部换行符（以及其他尾随空格），否则这些换行符将成为每行的一部分。

如果您的参数文件有一个标题，如您的示例所示，您有多种可能性跳过它。

例如，您可以一次读取所有行到列表中，然后迭代一个子集：

with open('parameters.txt', 'r') as my_file:
    all_lines = [line.rstrip() for line in my_file.readlines()]

# print(all_lines)

for line in all_lines[1:]:
    # print(line)

    my_function(line)

但是，这只会忽略标题。如果您不小心传递了错误的文件或内容无效的文件，这可能会造成麻烦。

最好检查文件的标题是否正确。您可以从上面展开代码：

with open('parameters.txt', 'r') as my_file:
    all_lines = [line.rstrip() for line in my_file.readlines()]

# print(all_lines)

if all_lines[0] != 'COM_PORTS':
    raise RuntimeError("file has wrong header")

for line in all_lines[1:]:
    # print(line)

    my_function(line)

或者你可以在循环中执行此操作，例如：

expect_header = True

with open('parameters.txt', 'r') as my_file:
    for line in my_file:
        stripped = line.rstrip()
        if expect_header:
            if stripped != 'COM_PORTS':
                raise RuntimeError("header of file is wrong")

            expect_header = False
            continue

        # print(stripped)

        my_function(stripped)

或者您可以使用生成器表达式来检查循环外的标头：

with open('parameters.txt', 'r') as my_file:
    all_lines = (line.rstrip() for line in my_file.readlines())

if next(all_lines) != 'COM_PORTS':
    raise RuntimeError("file has wrong header")

for line in all_lines:
    # print(line)

    my_function(line)

我可能更喜欢这最后一个，因为它有一个清晰的结构，没有神奇的数字（例如0和1，指的是哪个行是标题和分别跳过多少）并且它不需要全部读取一次排入内存。

但是，上面进一步将所有行读入列表的解决方案可能更好，如果您想对它们进行进一步处理，因为在这种情况下数据已经可用并且您不需要再次读取该文件。