猿问

将项目添加到列表列表 - Python

我希望使用此处的代码将域与其 DNS 解析器名称相匹配


当前 CSV 输出


domain1 dns1 dns2 dns3 \n domain2 dns1 dns2 dns3 \n etc

这是不正确的格式,因为它将所有域和 DNS 解析器添加到同一行,而不是基于新域的新行。由于换行符,它们仅由空白单元格分隔。相反,我希望将其编写如下,其中每个域及其 dns 解析器都写入其自己的单独行。


预期的 CSV 输出:


domain1 dns1 dns2 dns3

domain2 dns1 dns2 dns3

domain3 dns1 dns2 dns3

etc...

我希望 CSV 文件以正确的格式写出,并且使用我拥有的代码,每次将域传递给 def dns_resolver 时,它都应该迭代到新的列表索引。这样,每个域及其 dns 解析器都有自己的列表,因此在写入新的 CSV 文件时,每个域将打印在 CSV 文件中自己的行中。


该代码未正确迭代列表索引,因此不会将域及其 dns 名称添加到任何列表中。当它们全部写入同一个列表时,它工作正常,但它们全部写入同一行,这是不正确的。因此,我不会使用 1 个列表,而是使用一个列表列表,并将每个列表写入自己的列表,然后将每个列表写入 csv 文件,以便它们位于自己的行中。通常,域将从 csv 文件读入列表,但为此,我输入了 3 个值。


import dns.resolver 

import csv

import os

from os.path import dirname, abspath

r = 0


def dns_resolver(domain):

    server = []

    resolvers = []

    

    try:

        resolvers = dns.resolver.resolve(domain, 'NS')

        #dns_list.append(domain)

        for x in resolvers:

            #dns_list.append(x.target)

            #dns_list.append('\n') 

            server.append(str(x.target))

    except:

        server.append('did not resolve')

    finally:

        return (domain, *server)

            


# Read in all domains from csv file domains.csv & count how many domains there are listed

domain_list = ['google.com', 'facebook.com', 'github.com']

domain_amount = 0


with open(domainFName, 'r') as file:

    for line in csv.reader(file):

        name = (line)

        domain_list.append(line)

        domain_amount += 1


for first_domain in domain_list:

    for x in first_domain:

        outputWriter.writerow(dns_resolver(x))


慕姐4208626
浏览 94回答 1
1回答

qq_笑_17

您可以简单地让dns_resolver函数返回给定域的列表。这*server是将每个项目附加到列表中的简写。使用列表理解,将所有列表收集到列表列表中以写入 CSV。def dns_resolver(domain):    # do your dns resolution    # server = dns.resolver.resolve(domain, 'NS')    server = ["dns1", "dns2", "dns3", "dns4"]    return [domain, *server]        # Read in all domainsdomain_list = ['google.com', 'facebook.com', 'github.com']print([dns_resolver(d) for d in domain_list])输出:[   ['google.com', 'dns1', 'dns2', 'dns3', 'dns4'],   ['facebook.com', 'dns1', 'dns2', 'dns3', 'dns4'],   ['github.com', 'dns1', 'dns2', 'dns3', 'dns4']]
随时随地看视频慕课网APP

相关分类

Python
我要回答