Запрос IP-адреса DNS-сервера Python - PullRequest
0 голосов
/ 25 апреля 2018

Я пытаюсь получить IP-адрес DNS-сервера, используя python. Чтобы сделать это в командной строке Windows, я бы использовал

ipconfig -all

Как показано ниже:

enter image description here

Я хочу сделать то же самое, используя скрипт на Python. Есть ли способ извлечь эти значения? Мне удалось извлечь IP-адрес моего устройства, но IP-адрес DNS-сервера оказался более сложным.

1 Ответ

0 голосов
/ 14 декабря 2018

Мне недавно пришлось получить IP-адреса DNS-серверов, которые использовали набор кроссплатформенных хостов (linux, macOS, windows), вот как я в итоге это сделал, и я надеюсь, что это полезно:

#!/usr/bin/env python

import platform
import socket
import subprocess


def is_valid_ipv4_address(address):
    try:
        socket.inet_pton(socket.AF_INET, address)
    except AttributeError:  # no inet_pton here, sorry
        try:
            socket.inet_aton(address)
        except socket.error:
            return False
        return address.count('.') == 3
    except socket.error:  # not a valid address
        return False

    return True


def get_unix_dns_ips():
    dns_ips = []

    with open('/etc/resolv.conf') as fp:
        for cnt, line in enumerate(fp):
            columns = line.split()
            if columns[0] == 'nameserver':
                ip = columns[1:][0]
                if is_valid_ipv4_address(ip):
                    dns_ips.append(ip)

    return dns_ips


def get_windows_dns_ips():
    output = subprocess.check_output(["ipconfig", "-all"])
    ipconfig_all_list = output.split('\n')

    dns_ips = []
    for i in range(0, len(ipconfig_all_list)):
        if "DNS Servers" in ipconfig_all_list[i]:
            # get the first dns server ip
            first_ip = ipconfig_all_list[i].split(":")[1].strip()
            if not is_valid_ipv4_address(first_ip):
                continue
            dns_ips.append(first_ip)
            # get all other dns server ips if they exist
            k = i+1
            while k < len(ipconfig_all_list) and ":" not in ipconfig_all_list[k]:
                ip = ipconfig_all_list[k].strip()
                if is_valid_ipv4_address(ip):
                    dns_ips.append(ip)
                k += 1
            # at this point we're done
            break
    return dns_ips


def main():

    dns_ips = []

    if platform.system() == 'Windows':
        dns_ips = get_windows_dns_ips()
    elif platform.system() == 'Darwin':
        dns_ips = get_unix_dns_ips()
    elif platform.system() == 'Linux':
        dns_ips = get_unix_dns_ips()
    else:
        print("unsupported platform: {0}".format(platform.system()))

    print(dns_ips)
    return


if __name__ == "__main__":
    main()

Ресурсы, которые я использовал для создания этого скрипта:

https://stackoverflow.com/a/1325603

https://stackoverflow.com/a/4017219

Редактировать: Если у кого-то есть лучший способ сделать это, пожалуйста, поделитесь :)

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...