python sys.stdin.read() 来自 tail -f

Question

为什么 sys.stdin.read() 不从 tail -f 读取管道输入？

#!/usr/bin/env python
import sys
from geoip import geolite2
def iplookup(srcip):
        for ip in srcip.split("\n"):
                try:
                        print(geolite2.lookup(ip))
                except:
                        pass
source = sys.stdin.read()
iplookup(source)

tail -f/var/log/bleh.log | grep -oE '((1?[0-9][0-9]?|2[0-4][0-9]|25[0-5])\.){3}(1?[0 -9][0-9]?|2[0-4][0-9]|25[0-5])' | python mygeoip.py

Answer 1

你可以使用 fileinput :

import sys
from geoip import geolite2
import fileinput

def iplookup(srcip):
        for ip in srcip.split("\n"):
                try:
                        print(geolite2.lookup(ip))
                except:
                        pass

for line in fileinput.input():
    iplookup(line)

从好的方面来说，您的脚本也会自动接受文件名作为参数。