diff --git a/src/input/database.py b/src/input/database.py index 3a999e3..68ae4f5 100644 --- a/src/input/database.py +++ b/src/input/database.py @@ -101,10 +101,10 @@ def get_id(self): return self.lineEdit_dbid.displayText() def is_existing_id(self): - dbconn = self.get_dbconn() + dbconn, table_game, table_info = self.get_dbconn() condition = {'id': self.lineEdit_dbid.displayText()} - count = dbconn.count(table=dbconstant.TABLE_GAME, condition=condition) + count = dbconn.count(table=table_game, condition=condition) dbconn.close() return True if count > 0 else False