Skip to content

Commit

Permalink
fix: code for connecting database is modified
Browse files Browse the repository at this point in the history
  • Loading branch information
Yejining committed Feb 12, 2020
1 parent 5418059 commit 626f39c
Showing 1 changed file with 8 additions and 2 deletions.
10 changes: 8 additions & 2 deletions src/input/database.py
Original file line number Diff line number Diff line change
Expand Up @@ -66,7 +66,7 @@ def is_valid_id(self):
if len(self.lineEdit_dbid.displayText()) > 255: return False
return True

def is_existing_id(self):
def get_dbconn(self):
if self.custom_db_connectivity is True:
dbconn = MYSQL(
dbhost=self.table.item(0, 0).text(),
Expand All @@ -84,13 +84,19 @@ def is_existing_id(self):
dbcharset=dbconstant.CHARSET
)

return dbconn

def is_existing_id(self):
dbconn = self.get_dbconn()

condition = {'id': self.lineEdit_dbid.displayText()}
count = dbconn.count(table=dbconstant.TABLE, condition=condition)
dbconn.close()

return True if count > 0 else False

def save(self, data, parser, size):
dbconn = self.get_dbconn()
width, height = parser.get_card_size()
horizontal_margin, vertical_margin = parser.get_margins()

Expand Down Expand Up @@ -123,5 +129,5 @@ def save(self, data, parser, size):
'card_vertical_margin': vertical_margin
}

dbconstant.insert(table=dbconstant.TABLE, data=tuple)
dbconn.insert(table=dbconstant.TABLE, data=tuple)

0 comments on commit 626f39c

Please sign in to comment.