| Server IP : 88.222.222.138 / Your IP : 216.73.217.1 Web Server : LiteSpeed System : Linux id-dci-web1986.main-hosting.eu 5.14.0-611.26.1.el9_7.x86_64 #1 SMP PREEMPT_DYNAMIC Thu Jan 29 05:24:47 EST 2026 x86_64 User : u686484674 ( 686484674) PHP Version : 8.0.30 Disable Function : system, exec, shell_exec, passthru, mysql_list_dbs, ini_alter, dl, symlink, link, chgrp, leak, popen, apache_child_terminate, virtual, mb_send_mail MySQL : OFF | cURL : ON | WGET : ON | Perl : OFF | Python : OFF | Sudo : OFF | Pkexec : OFF Directory : /proc/self/root/opt/alt/python311/lib/python3.11/site-packages/pydig/ |
Upload File : |
import random
import subprocess
import logging
from .query_type import QueryType
logger = logging.getLogger(__name__)
class Resolver:
def __init__(
self,
executable='dig',
nameservers=None,
additional_args=None,
encoding='utf-8',
):
"""
Stores some customisable options into this resolver instance
"""
self.executable = executable
self.nameservers = nameservers or []
self.additional_args = additional_args or []
self.encoding = encoding
@property
def nameserver(self):
"""
Returns a random nameserver we should query against
"""
return random.choice(self.nameservers)
@staticmethod
def _execute(args):
"""
Calls out to subprocess with the passed in args
This method is normally mocked in tests
"""
logger.info('Executing subprocess.check_output({})'.format(repr(args))) # pragma: no cover
return subprocess.check_output(args) # pragma: no cover
def _args(self, domain, query_type):
"""
Builds up the final arguments to pass into subprocess
dig @1.1.1.1 example.com A +short
"""
yield self.executable
# Add in a random nameserver
if self.nameservers:
yield '@{}'.format(self.nameserver)
# The record we want to query
yield domain
# Our query type
yield query_type.name
# We only care about the result
yield '+short'
# Add in any additional args
yield from self.additional_args
def query(self, domain, query_type):
"""
Queries the resolver for a specific domain and query type
Returns a list of records
"""
domain = domain.lower()
query_type = QueryType.get(query_type)
args = list(self._args(domain, query_type))
output = self._execute(args).decode(self.encoding).strip()
return output.split('\n') if output else []