Fix to_target(duckdb) for stations

earthobservations · May 25, 2024 · 62458de · 62458de
1 parent 472a9f4
commit 62458de
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/CHANGELOG.rst b/CHANGELOG.rst
@@ -6,6 +6,7 @@ Development
 
 - Fix `dropna` argument for DWD Mosmix and DMO
 - Adjust DWD Mosmix and DMO kml reader to parse all parameters
+- Fix `to_target(duckdb)` for stations
 
 0.84.0 (15.05.2024)
 *******************

diff --git a/wetterdienst/core/timeseries/export.py b/wetterdienst/core/timeseries/export.py
@@ -5,6 +5,7 @@
 import json
 import logging
 from abc import abstractmethod
+from copy import copy
 from dataclasses import dataclass
 from urllib.parse import urlunparse
 
@@ -245,7 +246,12 @@ def to_target(self, target: str):
             log.info(f"Writing to DuckDB. database={database}, table={tablename}")
             import duckdb
 
-            df = self.df.with_columns(pl.col(Columns.DATE.value).dt.replace_time_zone(None)).to_pandas()
+            df = copy(self.df)
+
+            for column in (Columns.START_DATE.value, Columns.END_DATE.value, Columns.DATE.value):
+                if column in df.columns:
+                    df = df.with_columns(pl.col(column).dt.replace_time_zone(None))
+            df = df.to_pandas()
 
             connection = duckdb.connect(database=database, read_only=False)
             connection.register("origin", df)