Skip to content

Commit 071719b

Browse files
fix: create pandas dataframe with backward capabilities
1 parent 65a7863 commit 071719b

1 file changed

Lines changed: 9 additions & 0 deletions

File tree

influxdb_client/client/flux_csv_parser.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -251,6 +251,15 @@ def _prepare_data_frame(self):
251251

252252
# We have to create temporary DataFrame because we want to preserve default column values
253253
_temp_df = pd.DataFrame(self._data_frame_values)
254+
# This is for backward compatibles reason
255+
# In newer Pandas versions 'string' type will be 'str', in older versions 'string' will be 'object'
256+
# In newer Pandas versions 'time' will be 'datetime64[us, UTC]', in older versions 'time' will be 'datetime64[ns, UTC]'
257+
for column in _temp_df.columns:
258+
if _temp_df[column].dtype.name == 'str':
259+
_temp_df[column] = _temp_df[column].astype(object)
260+
if _temp_df[column].dtype.name == 'datetime64[us, UTC]':
261+
_temp_df[column] = _temp_df[column].astype('datetime64[ns, UTC]')
262+
254263
self._data_frame_values = []
255264

256265
# Custom DataFrame index

0 commit comments

Comments
 (0)