|
|
|
|
|
|
|
|
logging.info("single_sql: ", single_sql) |
|
|
logging.info("single_sql: ", single_sql) |
|
|
cursor.execute(single_sql) |
|
|
cursor.execute(single_sql) |
|
|
if cursor.rowcount == 0: |
|
|
if cursor.rowcount == 0: |
|
|
sql_res.append({"content": "\nTotal: 0\n No record in the database!"}) |
|
|
|
|
|
|
|
|
sql_res.append({"content": "No record in the database!"}) |
|
|
break |
|
|
break |
|
|
if self._param.db_type == 'mssql': |
|
|
if self._param.db_type == 'mssql': |
|
|
single_res = pd.DataFrame.from_records(cursor.fetchmany(self._param.top_n),columns = [desc[0] for desc in cursor.description]) |
|
|
single_res = pd.DataFrame.from_records(cursor.fetchmany(self._param.top_n),columns = [desc[0] for desc in cursor.description]) |
|
|
else: |
|
|
else: |
|
|
single_res = pd.DataFrame([i for i in cursor.fetchmany(self._param.top_n)]) |
|
|
single_res = pd.DataFrame([i for i in cursor.fetchmany(self._param.top_n)]) |
|
|
single_res.columns = [i[0] for i in cursor.description] |
|
|
single_res.columns = [i[0] for i in cursor.description] |
|
|
sql_res.append({"content": "\nTotal: " + str(cursor.rowcount) + "\n" + single_res.to_markdown()}) |
|
|
|
|
|
|
|
|
sql_res.append({"content": single_res.to_markdown()}) |
|
|
break |
|
|
break |
|
|
except Exception as e: |
|
|
except Exception as e: |
|
|
single_sql = self._regenerate_sql(single_sql, str(e), **kwargs) |
|
|
single_sql = self._regenerate_sql(single_sql, str(e), **kwargs) |