Skip to content
Snippets Groups Projects
Commit 592ba14b authored by Erik Johnston's avatar Erik Johnston
Browse files

Fix bugs in upgrade script.

Handle the case when there are colons in server_name. Handle http
exceptions more gracefully.
parent cb91ce5b
No related branches found
No related tags found
No related merge requests found
......@@ -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 {}
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment