Fix bugs in upgrade script.

Handle the case when there are colons in server_name. Handle http
exceptions more gracefully.
This commit is contained in:
Erik Johnston 2014-12-16 14:07:05 +00:00
parent cb91ce5bba
commit 592ba14b36

View File

@ -21,6 +21,7 @@ from syutil.jsonutil import encode_canonical_json
import argparse
import dns.resolver
import hashlib
import httplib
import json
import sqlite3
import syutil
@ -150,7 +151,6 @@ def get_key(server_name):
if ":" in server_name:
target, port = server_name.split(":")
targets.append((target, int(port)))
return
try:
answers = dns.resolver.query("_matrix._tcp." + server_name, "SRV")
for srv in answers:
@ -176,6 +176,10 @@ def get_key(server_name):
return verify_keys
except urllib2.URLError:
pass
except urllib2.HTTPError:
pass
except httplib.HTTPException:
pass
print "Failed to get keys for %s" % (server_name,)
return {}