从域名中查找国家并将结果放入文本文件

Question

我在下面有这个脚本如何检查域名的国家/地区。我想把每个国家代码放在一个特定的文件中。

例如：

FR ==> FR.txt
美国 ==> 美国.TXT

脚本输出是这样的：

domain.com [xx.xx.xx.xx]：美国

脚本：

#!/usr/bin/python

import socket
from geolite2 import geolite2

def origin(ip, domain_str, result):
    print("{0} [{1}]: {2}".format(domain_str.strip(), ip, result))

def getip(domain_str):
    ip = socket.gethostbyname(domain_str.strip())
    reader = geolite2.reader()      
    output = reader.get(ip)
    result = output['country']['iso_code']
    origin(ip, domain_str, result)

with open("hostnames.txt", "r") as ins:
    for domain_str in ins:
        try:
            getip(domain_str)
        except socket.error as msg:
            print("{0} [could not resolve]".format(domain_str.strip())) 
            if len(domain_str) > 2:
                subdomain = domain_str.split('.', 1)[1]
                try:
                    getip(subdomain)
                except:
                    continue

geolite2.close()

感谢您提供的任何帮助

Answer 1

def origin(ip, domain_str, result):
    file_name = "{}.txt".format(result)
    with open(file_name, 'a') as f:
        f.write("{}".format(domain_str.strip()))