File tree Expand file tree Collapse file tree 1 file changed +4
-3
lines changed Expand file tree Collapse file tree 1 file changed +4
-3
lines changed Original file line number Diff line number Diff line change 1
1
"""Parsing response from InfluxDB to FluxStructures or DataFrame."""
2
2
3
- import asyncio
4
3
import base64
5
4
import codecs
6
5
import csv as csv_parser
@@ -147,8 +146,10 @@ async def _parse_flux_response_async(self):
147
146
df = self ._prepare_data_frame ()
148
147
if not self ._is_profiler_table (metadata .table ):
149
148
yield df
150
- except (asyncio .exceptions .CancelledError , asyncio .TimeoutError ) as ce :
151
- ce .add_note ("Stream cancelled during read. Recommended: Check Influxdb client `timeout` setting." )
149
+ except BaseException as e :
150
+ e_type = type (e ).__name__
151
+ if "CancelledError" in e_type or "TimeoutError" in e_type :
152
+ e .add_note ("Stream cancelled during read. Recommended: Check Influxdb client `timeout` setting." )
152
153
raise
153
154
finally :
154
155
self ._close ()
You can’t perform that action at this time.
0 commit comments