python sys.stdin.read（）不需要的拆分

Question

我正在尝试制作一个脚本，可以将IP地址传送到该脚本，并使其返回geoip位置。 当我尝试此操作时，它看起来像一次将每个IP地址分割为1个字符，而在尝试运行iplookup（）时失败

#!/usr/bin/env python
import sys
from geoip import geolite2
def iplookup(srcip):
    for ip in srcip:
            print(ip)
            try:
                    print(geolite2.lookup(ip))
            except:
                    print("oops")
                    pass


source = sys.stdin.read()
iplookup(source)

回显数据样本（将来我希望来自tail -f fast.log）仅提取IP地址

[root@suricata py]# echo "05/20/2015-15:57:34.607470 [**] [1:2020702:2] ET DOS Bittorrent User-Agent inbound - possible DDOS [**] [Classification: Attempted Denial of Service] [Priority: 2] {TCP} 8.8.8.8:2039 -> 1.1.1.1:80" | 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

8
oops
.
oops
8
oops
.
oops
8
oops
.
oops
8
oops

等等...

Answer 1

sys.stdin.read（）为您提供一个字符串，如果您使用逗号分隔IP地址，请执行以下操作

source = sys.stdin.read()
for ip in source.split(","):
    print(geolite2.lookup(ip))

Answer 2

当您调用iplookup("stackoverflow.com") ，您的python函数将完全按照您的要求执行操作，即遍历输入数组的元素。 您的for循环告诉python浏览列表。

python中的字符串是字符列表。

因此，要么不要这样做（杀死循环），要么通过iplookup((source,))或iplookup([source])调用函数。