def _form_query_from_data(self, row, parsed):
"""Create a query from a row from the query table.
:param self: The current object
:type self: queryutils.databases.Database
:param row: The row fetched from the database
:type row: dict
:param parsed: Whether or not the row contains parsetree data
:type parsed: bool
:rtype: queryutils.query.Query
"""
d = { k:row[k] for k in row.keys() }
q = Query(row["text"], row["time"])
q.__dict__.update(d)
if parsed:
q.parsetree = ParseTreeNode.loads(row["parsetree"])
return q
def get_parsetrees(self):
"""Return the parsed queries from the parsetree table.
:param self: The current object
:type self: queryutils.databases.Database
:rtype: generator
"""
self.connect()
cursor = self.execute("SELECT parsetree, query_id FROM parsetrees")
for row in cursor.fetchall():
try:
p = ParseTreeNode.loads(row["parsetree"])
p.query_id = row["query_id"]
yield p
except ValueError as e:
print e
print parsetree
self.close()