使用Python从linux命令行读取

时间:2015-06-17 18:53:28

标签: python linux shell command-line

有没有办法读取进入命令行的数据,直接进入另一个Python脚本执行?

3 个答案:

答案 0 :(得分:4)

您需要从python脚本中读取stdin

import sys

data = sys.stdin.read()
print 'Data from stdin -', data

示例运行 -

$ date | python test.py
Data from stdin - Wed Jun 17 11:59:43 PDT 2015

答案 1 :(得分:0)

使用管道

x = input()

魔术

答案 2 :(得分:0)

这里是一个示例,可供将来参考,也对他人有帮助。
经过Python 2.6、2.7、3.6的测试

# coding=utf-8
"""
Read output from command line passed through a linux pipe
$ vmstat 1 | python read_cmd_output.py 10 3 10
This will warn when the column value breached the threshold for N time consecutively
"""

import sys
import re

def main():
    """ Main """
    values = []
    try:
        column, occurrence, threshold = sys.argv[1:]
        column = int(column)
        occurrence = int(occurrence)
        threshold = int(threshold)
    except ValueError:
        print('Usage: {0} <column> <occurrence> <threshold>'.format(sys.argv[0]))
        sys.exit(1)

    with sys.stdin:
        for line in iter(sys.stdin.readline, b''):
            line = line.strip()
            if re.match(r'\d', line):
                elems = re.split(r'\s+', line)
                values.append(elems[column-1])
                len(values) > occurrence and values.pop(0)
                nb_greater = len([x for x in values if int(x) > threshold])
                print(values, nb_greater, 'Warn' if nb_greater >= len(values) else '')

if __name__ == '__main__':
    try:
        main()
    except KeyboardInterrupt:
        print("\nUser interrupted the script")
        sys.exit(1)