e-data 2.0.2.dev136__py3-none-any.whl → 2.0.2.dev137__py3-none-any.whl

{e_data-2.0.2.dev136.dist-info → e_data-2.0.2.dev137.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: e-data
-Version: 2.0.2.dev136
+Version: 2.0.2.dev137
 Summary: Python library for managing spanish energy data from various web providers
 Author-email: VMG <vmayorg@outlook.es>
 License:                     GNU GENERAL PUBLIC LICENSE
@@ -697,7 +697,7 @@ Requires-Dist: pydantic<3,>=2.10
 Requires-Dist: python_dateutil<3,>=2.8
 Requires-Dist: Requests<3,>=2.31
 Requires-Dist: SQLAlchemy[asyncio]<3,>=2.0
-Requires-Dist: sqlmodel<0.0.45,>=0.0.22
+Requires-Dist: sqlmodel<0.1,>=0.0.45
 Requires-Dist: typer<1,>=0.12
 Requires-Dist: aiosqlite<1,>=0.21
 Dynamic: license-file

{e_data-2.0.2.dev136.dist-info → e_data-2.0.2.dev137.dist-info}/RECORD

@@ -1,4 +1,4 @@
-e_data-2.0.2.dev136.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+e_data-2.0.2.dev137.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
 edata/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 edata/cli.py,sha256=CsBYDggt0grkLb4cebrckOvJyHr9PaGTnlWqT4yjuFo,5215
 edata/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -7,7 +7,7 @@ edata/core/const.py,sha256=X9J2f7lt9jXJnghqz0JFi-J7USa3xlYvbuiki70eHNE,43
 edata/core/utils.py,sha256=uQX8GmNjuhof56jiXXoV-Fk-d590OuIoBJbM9ymZyV8,3194
 edata/database/__init__.py,sha256=HP_pDbGLzeMY9Ufm_Dps5cxzW_3YTonT8w8dstG558c,46
 edata/database/controller.py,sha256=bI3cE1Q4SK9U1Eep_61EhlmyVjgYH87F5lfBLhwSpmg,23211
-edata/database/models.py,sha256=tSkxf2OKgT1m2F1td3j3GtYcboUtSuYbhnHWQtpvJuY,5525
+edata/database/models.py,sha256=WWZHYgFBgSDqHORxw5z4Vip7HvorMcQbt9oCpV61NJA,5780
 edata/database/queries.py,sha256=XziDSG2V-PVFmvRkj8UrEvgrc9Qc-ShhnxgsPHIsyik,8304
 edata/database/utils.py,sha256=V-FJfyAh8YJBGHB85BbWtG3Ilq09pM9Gl-L86TffB4g,1062
 edata/database/migrations/__init__.py,sha256=P8VXZzfyceOV3GknYXif2HEMB8Bo4B7f7vKdIC02ZO0,1203
@@ -24,14 +24,14 @@ edata/services/bill_service.py,sha256=KAM78u7TpoVF-KTlB7bqj2Kp-rJERiPPoGTJCc-7rM
 edata/services/data_service.py,sha256=JJCdivjrwNt3aLhz3uMBWv06y-EL7UiK2G5EBr97l_8,19374
 edata/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 edata/tests/test_completion.py,sha256=-iyVt-juU3U1HqMuLUJXJthil2-n_4ddWiJq2GwnXTo,3551
-edata/tests/test_controller.py,sha256=E1Blvel2dtS8CiyxErP4V_wpzWlEcxTYA1gW1f8-Gg4,3449
+edata/tests/test_controller.py,sha256=YvilJ9cIpjhdGGtVJtlEevhRL7IHAVnU0At7AmxM5_8,3983
 edata/tests/test_datadis_connector.py,sha256=8Xet2M2eG-kMhiRDTCSH0R5nYO2oIpQTPOuTZKMKoLo,12538
 edata/tests/test_incremental.py,sha256=dn1LRgjBi2HuNYELPKDVdZQlS01hk4lsP9L-vyvtEHo,3479
 edata/tests/test_migrations.py,sha256=3Io2nokZVdv2xU9fCZCMIvwAwPVtGc0UivClSouP3pw,6418
 edata/tests/test_redata_connector.py,sha256=SBvxsma5t-qvz2drxNisNM2P1STTurqEs1TSDgwaPJ8,508
 edata/tests/test_services.py,sha256=Ya5cbADHj1Qsffb_IuMeDD0BViz8nqrgwazUWK0-hRc,8297
 edata/tests/test_utils.py,sha256=h6yBzT7j3eA42zbXJ-AK9lXcZkjCMf16wnob1oqGvUE,2307
-e_data-2.0.2.dev136.dist-info/METADATA,sha256=YZbWL0ZGXBHvfFBODBeXBwD8CcQHH5VXqYqY_NV42GE,48087
-e_data-2.0.2.dev136.dist-info/WHEEL,sha256=YVMoNqKzERt-wjUZwJ33xBGAwnFl-4cqbYkTtWa4itE,91
-e_data-2.0.2.dev136.dist-info/top_level.txt,sha256=Ez-fReWtUVTMcFuH0dzCnoKIgMBHgjLaBuba8QiKC7g,6
-e_data-2.0.2.dev136.dist-info/RECORD,,
+e_data-2.0.2.dev137.dist-info/METADATA,sha256=-7nOw3zSVBn4BVOgnmEXPt44pUE_qauv2ZZ2yjDjXt4,48084
+e_data-2.0.2.dev137.dist-info/WHEEL,sha256=YVMoNqKzERt-wjUZwJ33xBGAwnFl-4cqbYkTtWa4itE,91
+e_data-2.0.2.dev137.dist-info/top_level.txt,sha256=Ez-fReWtUVTMcFuH0dzCnoKIgMBHgjLaBuba8QiKC7g,6
+e_data-2.0.2.dev137.dist-info/RECORD,,

edata/database/models.py

@@ -1,6 +1,7 @@
 import typing
 from datetime import datetime as dt
 
+from pydantic import NaiveDatetime
 from sqlmodel import AutoString, Column, Field, Index, SQLModel, UniqueConstraint
 
 from edata.database.utils import PydanticJSON
@@ -27,8 +28,8 @@ class SupplyModel(SQLModel, table=True):
     cups: str = Field(default=None, primary_key=True)
     data: Supply = Field(sa_column=Column(PydanticJSON(Supply)))
     version: int = Field(default=1)
-    created_at: dt = Field(default_factory=_now, nullable=False)
-    updated_at: dt = Field(
+    created_at: NaiveDatetime = Field(default_factory=_now, nullable=False)
+    updated_at: NaiveDatetime = Field(
         default_factory=_now, nullable=False, sa_column_kwargs={"onupdate": dt.now}
     )
 
@@ -44,12 +45,12 @@ class ContractModel(SQLModel, table=True):
 
     id: int | None = Field(default=None, primary_key=True)
     cups: str = Field(foreign_key="supply.cups", index=True)
-    date_start: dt = Field(index=True)
+    date_start: NaiveDatetime = Field(index=True)
     data: Contract = Field(sa_column=Column(PydanticJSON(Contract)))
 
     version: int = Field(default=1)
-    created_at: dt = Field(default_factory=_now, nullable=False)
-    updated_at: dt = Field(
+    created_at: NaiveDatetime = Field(default_factory=_now, nullable=False)
+    updated_at: NaiveDatetime = Field(
         default_factory=_now, nullable=False, sa_column_kwargs={"onupdate": dt.now}
     )
 
@@ -71,13 +72,13 @@ class EnergyModel(SQLModel, table=True):
     id: int | None = Field(default=None, primary_key=True)
     cups: str = Field(foreign_key="supply.cups", index=True)
     delta_h: float
-    datetime: dt = Field(index=True)
+    datetime: NaiveDatetime = Field(index=True)
 
     data: Energy = Field(sa_column=Column(PydanticJSON(Energy)))
 
     version: int = Field(default=1)
-    created_at: dt = Field(default_factory=_now, nullable=False)
-    updated_at: dt = Field(
+    created_at: NaiveDatetime = Field(default_factory=_now, nullable=False)
+    updated_at: NaiveDatetime = Field(
         default_factory=_now, nullable=False, sa_column_kwargs={"onupdate": dt.now}
     )
 
@@ -93,13 +94,13 @@ class PowerModel(SQLModel, table=True):
 
     id: int | None = Field(default=None, primary_key=True)
     cups: str = Field(foreign_key="supply.cups", index=True)
-    datetime: dt = Field(index=True)
+    datetime: NaiveDatetime = Field(index=True)
 
     data: Power = Field(sa_column=Column(PydanticJSON(Power)))
 
     version: int = Field(default=1)
-    created_at: dt = Field(default_factory=_now, nullable=False)
-    updated_at: dt = Field(
+    created_at: NaiveDatetime = Field(default_factory=_now, nullable=False)
+    updated_at: NaiveDatetime = Field(
         default_factory=_now, nullable=False, sa_column_kwargs={"onupdate": dt.now}
     )
 
@@ -120,14 +121,14 @@ class StatisticsModel(SQLModel, table=True):
 
     id: int | None = Field(default=None, primary_key=True)
     cups: str = Field(foreign_key="supply.cups", index=True)
-    datetime: dt = Field(index=True)
+    datetime: NaiveDatetime = Field(index=True)
     type: typing.Literal["day", "month"] = Field(index=True, sa_type=AutoString)
     complete: bool = Field(False)
     data: Statistics = Field(sa_column=Column(PydanticJSON(Statistics)))
 
     version: int = Field(default=1)
-    created_at: dt = Field(default_factory=_now, nullable=False)
-    updated_at: dt = Field(
+    created_at: NaiveDatetime = Field(default_factory=_now, nullable=False)
+    updated_at: NaiveDatetime = Field(
         default_factory=_now, nullable=False, sa_column_kwargs={"onupdate": dt.now}
     )
 
@@ -137,12 +138,12 @@ class PVPCModel(SQLModel, table=True):
     __tablename__ = "pvpc"  # type: ignore
 
     id: int | None = Field(default=None, primary_key=True)
-    datetime: dt = Field(index=True, unique=True)
+    datetime: NaiveDatetime = Field(index=True, unique=True)
     data: EnergyPrice = Field(sa_column=Column(PydanticJSON(EnergyPrice)))
 
     version: int = Field(default=1)
-    created_at: dt = Field(default_factory=_now, nullable=False)
-    updated_at: dt = Field(
+    created_at: NaiveDatetime = Field(default_factory=_now, nullable=False)
+    updated_at: NaiveDatetime = Field(
         default_factory=_now, nullable=False, sa_column_kwargs={"onupdate": dt.now}
     )
 
@@ -163,14 +164,14 @@ class BillModel(SQLModel, table=True):
 
     id: int | None = Field(default=None, primary_key=True)
     cups: str = Field(foreign_key="supply.cups", index=True)
-    datetime: dt = Field(index=True)
+    datetime: NaiveDatetime = Field(index=True)
     type: typing.Literal["hour", "day", "month"] = Field(index=True, sa_type=AutoString)
     complete: bool = Field(False)
     confhash: str
     data: Bill = Field(sa_column=Column(PydanticJSON(Bill)))
 
     version: int = Field(default=1)
-    created_at: dt = Field(default_factory=_now, nullable=False)
-    updated_at: dt = Field(
+    created_at: NaiveDatetime = Field(default_factory=_now, nullable=False)
+    updated_at: NaiveDatetime = Field(
         default_factory=_now, nullable=False, sa_column_kwargs={"onupdate": dt.now}
     )

edata/tests/test_controller.py

@@ -6,6 +6,8 @@ from datetime import datetime, timedelta
 
 import pytest
 import pytest_asyncio
+from sqlalchemy import DateTime
+from sqlmodel import SQLModel
 
 from edata.database.controller import EdataDB
 from edata.models import Bill, Energy, Supply
@@ -106,3 +108,16 @@ async def test_concurrent_first_calls_do_not_race_index_creation(db: EdataDB) ->
             "SELECT name FROM sqlite_master WHERE name='ix_energy_cups_datetime'"
         )
         assert result.first() is not None
+
+
+def test_datetime_columns_are_naive() -> None:
+    # the supply timezone is unknown, so datetimes are stored as-is (naive);
+    # plain ``datetime`` fields map to UTC-aware columns on sqlmodel>=0.0.45
+    columns = [
+        column
+        for table in SQLModel.metadata.sorted_tables
+        for column in table.columns
+        if isinstance(column.type, DateTime)
+    ]
+    assert len(columns) == 20
+    assert all(column.type.timezone is False for column in columns)