Method execute_sql_api

methods/ReFoRCE/sql.py:133–149

(self, sql_query, ex_id, save_path=None, api="sqlite", max_len=30000, sqlite_path=None, timeout=300)

    def execute_sql_api(self, sql_query, ex_id, save_path=None, api="sqlite", max_len=30000, sqlite_path=None):
        if api == "bigquery":
            return self.exec_sql_bq(sql_query, save_path, max_len)
        elif api == "snowflake":
            if ex_id not in self.conns.keys():
                self.start_db_sf(ex_id)
            return self.exec_sql_sf(sql_query, save_path, max_len, ex_id)
        elif api == "sqlite":
            if sqlite_path not in self.conns.keys():
                self.start_db_sqlite(sqlite_path)
            return self.exec_sql_sqlite(sql_query, save_path, max_len, sqlite_path)