parouby
/
Pamhyr2
mirror of https://gitlab.com/pamhyr/pamhyr2
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Pamhyr2: Fix some type compare.

setup.py
Pierre-Antoine Rouby 2023-11-22 15:47:57 +01:00
parent 2af9b0018c
commit 01fd7dbec8
2 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/Model/Tools/PamhyrDB.py
View File

@ -115,9 +115,9 @@ class SQLSubModel(object):
def _db_format(self, value): def _db_format(self, value):
# Replace ''' by ''' to preserve SQL injection # Replace ''' by ''' to preserve SQL injection
if value is str: if type(value) is str:
value = value.replace("'", "'") value = value.replace("'", "'")
elif value is bool: elif type(value) is bool:
value = 'TRUE' if value else 'FALSE' value = 'TRUE' if value else 'FALSE'
return value return value

10
src/tools.py
View File

@ -297,7 +297,7 @@ class SQL(object):
def _fetch_tuple(self, tup): def _fetch_tuple(self, tup):
res = [] res = []
for v in tup: for v in tup:
if v is str: if type(v) is str:
v = self._fetch_string(v) v = self._fetch_string(v)
res.append(v) res.append(v)
@ -306,9 +306,9 @@ class SQL(object):
def _fetch_list(self, lst): def _fetch_list(self, lst):
res = [] res = []
for v in lst: for v in lst:
if v is str: if type(v) is str:
v = self._fetch_string(v) v = self._fetch_string(v)
elif v is tuple: elif type(v) is tuple:
v = self._fetch_tuple(v) v = self._fetch_tuple(v)
res.append(v) res.append(v)
@ -321,9 +321,9 @@ class SQL(object):
value = res.fetchall() value = res.fetchall()
res = value res = value
if value is list: if type(value) is list:
res = self._fetch_list(value) res = self._fetch_list(value)
elif value is tuple: elif type(value) is tuple:
res = self._fetch_tuple(value) res = self._fetch_tuple(value)
return res return res