Improved HttpProxy validation routine

Validation routine for HttpProxy updated to use the newly introduced isSocketAddress() function
This commit is contained in:
Junaid Loonat 2016-04-26 23:32:48 +02:00
parent 3e72aa5da5
commit 3a2b603f2f
1 changed files with 7 additions and 10 deletions

View File

@ -35,6 +35,7 @@ import socket
from . import ircutils, registry, utils
from .utils import minisix
from .utils.net import isSocketAddress
from .version import version
from .i18n import PluginInternationalization
_ = PluginInternationalization()
@ -1179,17 +1180,13 @@ class HttpProxy(registry.String):
def setValue(self, v):
proxies = {}
if v != "":
# TODO: improve checks
if ':' not in v:
if isSocketAddress(v):
proxies = {
'http': v,
'https': v
}
else:
self.error()
try:
int(v.rsplit(':', 1)[1])
except ValueError:
self.error()
proxies = {
'http': v,
'https': v
}
proxyHandler = ProxyHandler(proxies)
proxyOpenerDirector = build_opener(proxyHandler)
install_opener(proxyOpenerDirector)