fix: database checking id

DVL-Sejong · Feb 17, 2020 · 80410eb · 80410eb
1 parent a2966d4
commit 80410eb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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