def append(self, q, params=None, returns=None, obj=None): result_data_contents = ["REST"] if in_ipnb(): result_data_contents += ["row", "graph"] statement = { "statement": q, "parameters": params, "resultDataContents": result_data_contents, } self.statements.append(statement) self.references.append({ "statement": statement, "returns": returns, "obj": obj, })
def append(self, q, params=None, returns=None, obj=None, data_contents=None): result_data_contents = ["REST"] if in_ipnb() or data_contents is True: result_data_contents += ["row", "graph"] include_stats = True elif data_contents and data_contents.lower() in ["row", "graph"]: include_stats = True result_data_contents.append(data_contents.lower()) else: include_stats = False statement = { "statement": q, "parameters": params, "includeStats": include_stats, "resultDataContents": result_data_contents, } self.statements.append(statement) self.references.append({ "statement": statement, "returns": returns, "obj": obj, })