从命令行、Linux 中使用函数执行 python 脚本

Question

我有一个名为convertImage.py的python文件，在文件中我有一个脚本可以根据我的喜好转换图像，整个转换脚本设置在一个名为convertFile(fileName)的函数内

现在我的问题是我需要执行这个python脚本从linux命令行，同时传递convertFile(fileName)函数。

示例：

 linux user$: python convertImage.py convertFile(fileName)

这应该执行传递适当函数的 python 脚本。

示例：

def convertFile(fileName):

    import os, sys
    import Image
    import string

    splitName = string.split(fileName, "_")
    endName = splitName[2]
    splitTwo = string.split(endName, ".")
    userFolder = splitTwo[0]

    imageFile = "/var/www/uploads/tmp/"+fileName

    ...rest of the script...

    return

执行此 python 脚本并将文件名从 liunx 命令行正确传递给函数的正确方法是什么？

提前致谢

Answer 1

这

if __name__ == "__main__":
    command= " ".join( sys.argv[1:] )
    eval( command )

这会起作用。但这非常危险。

您确实需要考虑您的命令行语法是什么。您需要考虑为什么要打破长期以来为程序指定参数的 Linux 标准。

例如，您应该考虑删除示例中无用的 ()。改为这样吧。

python convertImage.py convertFile fileName

然后，您可以 - 只需很少的工作 - 使用 argparse 来获取命令（“convertFile”）和参数（“fileName”）并在标准 Linux 命令行语法中工作。

function_map = { 
    'convertFile': convertFile,
    'conv': convertFile,
}
parser = argparse.ArgumentParser()
parser.add_argument( 'command', nargs=1 )
parser.add_argument( 'fileName', nargs='+' )
args= parser.parse_args()
function = function_map[args.command]
function( args.fileName )

Answer 2

快速而肮脏的方法：

linux user$: python convertImage.py convertFile fileName

然后在convertImage.py中

if __name__ == '__main__':
    import sys
    function = getattr(sys.modules[__name__], sys.argv[1])
    filename = sys.argv[2]
    function(filename)

更复杂的方法是使用 argparse（适用于 2.7 或 3.2+）或 optparse。

Answer 3

创建脚本的顶级可执行部分，用于解析命令行参数，然后将其传递给调用中的函数，如下所示：

import os, sys
#import Image
import string


def convertFile(fileName):
    splitName = string.split(fileName, "_")
    endName = splitName[2]
    splitTwo = string.split(endName, ".")
    userFolder = splitTwo[0]

    imageFile = "/var/www/uploads/tmp/"+fileName

    print imageFile     # (rest of the script)

    return


if __name__ == '__main__':
    filename = sys.argv[1]
    convertFile(filename)

然后，从 shell，

$ convertImage.py the_image_file.png
/var/www/uploads/tmp/the_image_file.png