def retrieve_result(self, query_string):
# TODO: this method is hopefully temporary until InfluxDB /api/v2/query API simplifies passing
# mimic default dialect
dialect = Dialect(header=True,
delimiter=",",
comment_prefix="#",
annotations=["datatype", "group", "default"],
date_time_format="RFC3339")
query = Query(query=query_string, dialect=dialect, extern=self._extern)
try:
response = self.query_service.post_query(
org=self.client.org,
query=query,
async_req=False,
_preload_content=False,
_return_http_data_only=False)
_parser = FluxCsvParser(
response=response,
serialization_mode=FluxSerializationMode.tables)
list(_parser.generator())
tables = _parser.tables
except rest.ApiException as error:
# TODO: log error
print('Error while retrieving data:', error)
tables = []
return tables
def _create_query(self,
query,
dialect=default_dialect,
params: dict = None):
query_options = self._get_query_options()
profilers = query_options.profilers if query_options is not None else None
q = Query(query=query,
dialect=dialect,
extern=QueryApi._build_flux_ast(params, profilers))
if profilers:
print("\n===============")
print("Profiler: query")
print("===============")
print(query)
return q
def _create_query(self,
query,
dialect=default_dialect,
params: dict = None,
**kwargs):
query_options = self._get_query_options()
profilers = query_options.profilers if query_options is not None else None
q = Query(query=query,
dialect=dialect,
extern=_BaseQueryApi._build_flux_ast(params, profilers))
if profilers:
print("\n===============")
print("Profiler: query")
print("===============")
print(query)
if kwargs.get('dataframe_query', False):
MissingPivotFunction.print_warning(query)
return q
def _create_query(query, dialect=default_dialect, params: dict = None):
created = Query(query=query,
dialect=dialect,
extern=QueryApi._build_flux_ast(params))
return created
def _create_query(query, dialect=default_dialect):
created = Query(query=query, dialect=dialect)
return created