SQLite3-Datensatzüberprüfung

Question

Ich möchte, dass mein Code eine if-Anweisung verwendet, um zu überprüfen, in welcher Benutzergruppe sich ein Benutzer befindet, und verschiedene Funktionen für jeden Benutzer aufzurufen. Meine aktuellen Code ist wie folgt:

c.execute('SELECT * from users WHERE username=? AND password =?', 
       (username_input, password_input)) 
    if c.fetchone() is not None: 
     c.execute('SELECT usergroup from users WHERE username=? AND password=?', 
           (username_input, password_input)) 
     user_group = c.fetchone() 
     for (usergroup,) in c: 
      if user_group == 1: 
       App.admin_login_successful(self) 
      elif user_group == 2: 
       App.user_login_successful(self) 
      else: 
       App.user_login_successful(self) 
       # This shouldn't happen, as all records should theoretically contain a value for usergroup. 

Was soll ich den Vergleich der Arbeit tun, um, wie meine Platten einen Wert enthalten, die entweder 1 oder 2 ist, aber die Validierung nicht funktioniert.

Answer 1

Wiederholen Sie nicht die gleiche SQL-Abfrage mehr als einmal.

funcmap = {1: self.admin_login_successful, 2: self.user_login_successful} 
c.execute('SELECT usergroup from users WHERE username=? AND password=?', 
      (username_input, password_input)) 
for (usergroup,) in c: 
    funcmap[usergroup]() 

Wenn die Abfrage keine Benutzergruppen, dann ist die for-Schleife nicht seinen Körper auszuführen. Wenn usergroup nicht 1 oder 2 ist, wird funcmap[usergroup] eine KeyError auslösen. Wie Sie sagen, diese sollte nicht passieren, aber wenn es die KeyError wird Sie wissen lassen. Vielleicht möchten Sie eine try..except KeyError Anweisung verwenden, um funcmap[usergroup]() den Fehler auf andere Weise (Protokollierung, besser spezifische Fehlermeldung vor dem Beenden, etc.)

Beachten Sie, dass in Abhängigkeit davon, zu handhaben, wie Sie die sqlite3 Verbindung instanziiert und wie Sie eingestellt up die Datenbanktabelle, usergroup kann eine Ganzzahl oder eine Zeichenfolge sein. Wenn es eine Zeichenfolge zurückgibt, möchten Sie möglicherweise eine weitere Frage dazu stellen, wie Sie usergroup als Ganzzahl abrufen können. Alternativ können Sie funcmap ändern Strings wie die dict Schlüssel zu erwarten:

funcmap = {'1': self.admin_login_successful, '2': self.user_login_successful}