paulin/EasyGA
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Added type checking if the query is not a list or nested list

Browse Source
This commit is contained in:
SimpleArt
2020-11-22 22:32:25 -05:00
parent 4d04b7daab
commit 03debc145c
1 changed files with 3 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/database/sql_database.py
View File

@ -23,11 +23,11 @@ class SQL_Database:
query = method(self, config_id) query = method(self, config_id)
# Unpack elements if they are lists with only 1 element # Unpack elements if they are lists with only 1 element
if len(query[0]) == 1: if type(query[0]) in (list, tuple) and len(query[0]) == 1:
query = [i[0] for i in query] query = [i[0] for i in query]
# Unpack list if it is a list with only 1 element # Unpack list if it is a list with only 1 element
if len(query) == 1: if type(query) in (list, tuple) and len(query) == 1:
query = query[0] query = query[0]
return query return query
@ -212,9 +212,7 @@ class SQL_Database:
cur = self.conn.cursor() cur = self.conn.cursor()
cur.execute(query) cur.execute(query)
query_data = cur.fetchone() return cur.fetchone()
return query_data
def past_runs(self): def past_runs(self):