From 27631f1157226bd9ca4d0dbfb6a59c7656e7e361 Mon Sep 17 00:00:00 2001
From: Lukas Fleischer <lfleischer@archlinux.org>
Date: Thu, 4 Aug 2016 21:00:50 +0200
Subject: git-interface: Do not use rowcount

Avoid using Cursor.rowcount to obtain the number of rows returned by a
SELECT statement as this is not guaranteed to be supported by every
database engine.

Signed-off-by: Lukas Fleischer <lfleischer@archlinux.org>
---
 git-interface/git-auth.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

(limited to 'git-interface/git-auth.py')

diff --git a/git-interface/git-auth.py b/git-interface/git-auth.py
index ebdc75c..45fd577 100755
--- a/git-interface/git-auth.py
+++ b/git-interface/git-auth.py
@@ -40,10 +40,11 @@ cur = conn.execute("SELECT Users.Username, Users.AccountTypeID FROM Users " +
                    "WHERE SSHPubKeys.PubKey = ? AND Users.Suspended = 0",
                    (keytype + " " + keytext,))
 
-if cur.rowcount != 1:
+row = cur.fetchone()
+if not row or cur.fetchone():
     exit(1)
 
-user, account_type = cur.fetchone()
+user, account_type = row
 if not re.match(username_regex, user):
     exit(1)
 
-- 
cgit v1.2.3-54-g00ecf