Skip to content

Commit

Permalink
fix: remove debugging code from sqlite wrapper for gobot
Browse files Browse the repository at this point in the history
  • Loading branch information
yoptar committed Sep 6, 2019
1 parent b8809d0 commit 95dd339
Showing 1 changed file with 6 additions and 12 deletions.
18 changes: 6 additions & 12 deletions deeppavlov/core/data/sqlite_database.py
Original file line number Diff line number Diff line change
Expand Up @@ -86,18 +86,12 @@ def _check_if_table_exists(self):

def _search(self, kv=None, order_by=None, order=''):
order_expr = f" ORDER BY {order_by} {order}" if order_by else ''
try:
if kv:
keys, values = zip(*kv.items())
where_expr = " AND ".join(f"{k}=?" for k in keys)
sql = f"SELECT * FROM {self.tname} WHERE {where_expr}" + order_expr
self.cursor.execute(sql, values)
else:
sql = f"SELECT * FROM {self.tname}" + order_expr
self.cursor.execute(sql)
except:
print(sql)
raise
if kv:
keys, values = zip(*kv.items())
where_expr = " AND ".join(f"{k}=?" for k in keys)
self.cursor.execute(f"SELECT * FROM {self.tname} WHERE {where_expr}" + order_expr, values)
else:
self.cursor.execute(f"SELECT * FROM {self.tname}" + order_expr)
return [self._wrap_selection(s) for s in self.cursor.fetchall()]

def _wrap_selection(self, selection):
Expand Down

0 comments on commit 95dd339

Please sign in to comment.