Signed-off-by: Johannes Löthberg <johannes@kyriasis.com> --- git-interface/git-auth.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/git-interface/git-auth.py b/git-interface/git-auth.py index 8682275..0918f4a 100755 --- a/git-interface/git-auth.py +++ b/git-interface/git-auth.py @@ -41,18 +41,25 @@ db = mysql.connector.connect(host=aur_db_host, user=aur_db_user, unix_socket=aur_db_socket, buffered=True) cur = db.cursor() -cur.execute("SELECT Username FROM Users WHERE SSHPubKey = %s " + +cur.execute("SELECT Username, AccountTypeID FROM Users WHERE SSHPubKey = %s " + "AND Suspended = 0", (keytype + " " + keytext,)) if cur.rowcount != 1: exit(1) -user = cur.fetchone()[0] +user, account_type = cur.fetchone()[0] if not re.match(username_regex, user): exit(1) + +if account_type > 1: + privileged = True +else: + privileged = False + env_vars = { 'AUR_USER': user, + 'AUR_PRIVILEGED': privileged, } key = keytype + ' ' + keytext -- 2.4.4