diff --git a/db.py b/db.py index e260539..ace7417 100644 --- a/db.py +++ b/db.py @@ -43,8 +43,9 @@ def AddVPNImage(name, token): def GetVPNImage(token): Connect() with GetCur() as cur: - return cur.execute(""" + cur.execute(""" SELECT image_name FROM vpn WHERE token = %s - """,(token,)).fetchone() + """,(token,)) + return cur.fetchone() \ No newline at end of file