python - 如何为python代码文件创建我们自己的linux命令

Question

我是Python新手。作为我项目的一部分，我正在尝试为我已有的 python 文件创建一个 linux 命令。例如我有一个python文件example.py，

with open('file.txt','r') as f:
    for i in f:
        print i
with open('file.txt','r') as f:     #for different reasons I am opening file again
    for i in f:
        print i

在这里，我尝试创建类似 $example --print file.txt 的命令。这意味着我从命令本身给出输入文件。然后它必须打印输入文件的内容。请帮助我如何实现这一目标。提前致谢。

Answer 1

您需要将解释器添加到脚本的头部，并使用 sys.env 来访问输入参数:

#!/usr/bin/python

import sys, os;

# check so we have enough arguments
if len(sys.argv) < 2:
  printf "You need to supply at least two arguments."
  exit(-1)

# check so the file exists
if os.path.exists(sys.argv[1]):
  # try to open the filename supplied in the second argument
  # the sys.argv[0] always points to the name of the script
  with open(sys.argv[1], 'r') as f:
    for i in f:
      print i
else:
  print 'The file %s does not exists' %(sys.argv[1])

请注意，您可能需要将 /usr/bin/python 更改为安装 python 二进制文件的路径。您可以通过发出以下命令找出它的位置:

whereis python

现在您应该能够像这样运行命令:

./command file.txt

请注意，这里我假设您与 python 脚本位于同一文件夹中。您还可以将脚本移动到 $PATH 环境变量中的文件夹中，以便您可以从任何地方访问它。例如，您可以将其移动到 /usr/local/bin:

mv command /usr/local/bin/.

然后您应该能够使用以下命令从任何文件夹运行它:

command file.txt

您需要做的最后一件事是确保脚本可执行:

chmod +x command