From 1c4883edbab3f6a59598f54c894f62fa8590cf28 Mon Sep 17 00:00:00 2001 From: simo Date: Sun, 18 Dec 2005 00:17:58 +0000 Subject: More descriptive tupkg error message Catch sql server connection death --- tupkg/client/tupkg | 3 +++ tupkg/server/tupkgs | 14 +++++++++----- 2 files changed, 12 insertions(+), 5 deletions(-) (limited to 'tupkg') diff --git a/tupkg/client/tupkg b/tupkg/client/tupkg index c4806c9f..021e647b 100755 --- a/tupkg/client/tupkg +++ b/tupkg/client/tupkg @@ -87,6 +87,9 @@ class ClientSocket: reply = self.readMsg(1) if reply['result'] == ["PASS"]: return 1 + elif reply['result'] == ["SQLERR"]: + print "SQL server-side error" + return 0 else: return 0 diff --git a/tupkg/server/tupkgs b/tupkg/server/tupkgs index ca9ab26d..05d60241 100755 --- a/tupkg/server/tupkgs +++ b/tupkg/server/tupkgs @@ -119,11 +119,15 @@ class ClientSocket(threading.Thread): m = md5.new() m.update(authdata['password'][0]) encpw = m.hexdigest() - q.execute("SELECT ID, Suspended, AccountTypeID FROM Users WHERE Username = '"+ - MySQLdb.escape_string(authdata['username'][0])+ - "' AND Passwd = '"+ - MySQLdb.escape_string(encpw)+ - "'") + try: + q.execute("SELECT ID, Suspended, AccountTypeID FROM Users WHERE Username = '"+ + MySQLdb.escape_string(authdata['username'][0])+ + "' AND Passwd = '"+ + MySQLdb.escape_string(encpw)+ + "'") + except MySQLdb.OperationalError: + self.sendMsg("result=SQLERR") + return 0 if q.rowcount == 0: self.sendMsg("result=FAIL") return 0 -- cgit v1.2.3-24-g4f1b