added inline documentation for new dns methods

This commit is contained in:
Nathan Fritz 2011-08-18 01:04:01 -07:00
parent fb78bf0996
commit f75b6bf955
2 changed files with 24 additions and 0 deletions

View file

@ -141,6 +141,14 @@ class ClientXMPP(BaseXMPP):
use_tls=use_tls, reattempt=reattempt) use_tls=use_tls, reattempt=reattempt)
def get_dns_records(self, domain, port=None): def get_dns_records(self, domain, port=None):
"""
Get the DNS records for a domain.
Overridden XMLStream.get_dns_records to use SRV.
Arguments:
domain -- The domain in question.
port -- If the results don't include a port, use this one.
"""
if port is None: if port is None:
port = self.default_port port = self.default_port
if DNSPYTHON: if DNSPYTHON:

View file

@ -657,6 +657,13 @@ class XMLStream(object):
return False return False
def get_dns_records(self, domain, port=None): def get_dns_records(self, domain, port=None):
"""
Get the DNS records for a domain.
Arguments:
domain -- The domain in question.
port -- If the results don't include a port, use this one.
"""
if port is None: if port is None:
port = self.default_port port = self.default_port
if DNSPYTHON: if DNSPYTHON:
@ -673,6 +680,15 @@ class XMLStream(object):
return [((domain, port), 0, 0)] return [((domain, port), 0, 0)]
def pick_dns_answer(self, domain, port=None): def pick_dns_answer(self, domain, port=None):
"""
Pick a server and port from DNS answers.
Gets DNS answers if none available.
Removes used answer from available answers.
Arguments:
domain -- The domain in question.
port -- If the results don't include a port, use this one.
"""
if not self.dns_answers: if not self.dns_answers:
self.dns_answers = self.get_dns_records(domain, port) self.dns_answers = self.get_dns_records(domain, port)
addresses = {} addresses = {}