ngits-iot-db 0.1.0__tar.gz

ngits_iot_db-0.1.0/LICENSE

@@ -0,0 +1,28 @@
+BSD 3-Clause License
+
+Copyright (c) 2025, NG IT Services Sp. z o. o.
+
+Redistribution and use in source and binary forms, with or without
+modification, are permitted provided that the following conditions are met:
+
+1. Redistributions of source code must retain the above copyright notice, this
+   list of conditions and the following disclaimer.
+
+2. Redistributions in binary form must reproduce the above copyright notice,
+   this list of conditions and the following disclaimer in the documentation
+   and/or other materials provided with the distribution.
+
+3. Neither the name of the copyright holder nor the names of its
+   contributors may be used to endorse or promote products derived from
+   this software without specific prior written permission.
+
+THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
+FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
+SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
+CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
+OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

ngits_iot_db-0.1.0/MANIFEST.in

@@ -0,0 +1,13 @@
+include README.md
+include LICENSE
+include pyproject.toml
+include requirements.txt
+include alembic.ini
+exclude CLAUDE.md
+recursive-include alembic *.py *.mako
+recursive-include models *.py
+recursive-include alembic/versions *.py
+global-exclude *.pyc
+global-exclude __pycache__
+global-exclude .git*
+global-exclude .DS_Store

ngits_iot_db-0.1.0/PKG-INFO

@@ -0,0 +1,67 @@
+Metadata-Version: 2.4
+Name: ngits-iot-db
+Version: 0.1.0
+Summary: SQLAlchemy models for IoT sensor data storage with Alembic migrations
+Author-email: "NG IT Services Sp. z o. o." <biuro@ngits.pl>
+Keywords: iot,sqlalchemy,database,models,sensors
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: BSD License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Topic :: Database
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: SQLAlchemy>=2.0.0
+Requires-Dist: alembic>=1.13.0
+Requires-Dist: psycopg2-binary>=2.9.0
+Provides-Extra: dev
+Requires-Dist: black>=25.1.0; extra == "dev"
+Requires-Dist: isort>=5.13.0; extra == "dev"
+Requires-Dist: mypy>=1.0.0; extra == "dev"
+Requires-Dist: pytest>=7.0.0; extra == "dev"
+Requires-Dist: pytest-cov>=4.0.0; extra == "dev"
+Dynamic: license-file
+
+database uri
+============
+
+You can set database uri by setting `IOT_DATABASE_URL` variable.
+
+```shell
+export IOT_DATABASE_URL=postgresql://postgres:postgres@localhost:5432/iot_local
+```
+
+
+make migrations
+===============
+
+```shell
+alembic revision --autogenerate -m "Migration message."
+```
+
+upgrade
+=======
+
+```shell
+# upgrade all
+alembic upgrade head
+
+# upgrade all to specific migration
+alembic upgrade <revision_id>
+```
+
+downgrade
+=========
+
+```shell
+# downgrade all
+alembic downgrade base
+
+# downgrade all to specific migration
+alembic downgrade <revision_id>
+```

ngits_iot_db-0.1.0/README.md

@@ -0,0 +1,38 @@
+database uri
+============
+
+You can set database uri by setting `IOT_DATABASE_URL` variable.
+
+```shell
+export IOT_DATABASE_URL=postgresql://postgres:postgres@localhost:5432/iot_local
+```
+
+
+make migrations
+===============
+
+```shell
+alembic revision --autogenerate -m "Migration message."
+```
+
+upgrade
+=======
+
+```shell
+# upgrade all
+alembic upgrade head
+
+# upgrade all to specific migration
+alembic upgrade <revision_id>
+```
+
+downgrade
+=========
+
+```shell
+# downgrade all
+alembic downgrade base
+
+# downgrade all to specific migration
+alembic downgrade <revision_id>
+```

ngits_iot_db-0.1.0/alembic/env.py

@@ -0,0 +1,83 @@
+import os
+from logging.config import fileConfig
+
+from sqlalchemy import engine_from_config, pool
+
+from alembic import context
+
+# this is the Alembic Config object, which provides
+# access to the values within the .ini file in use.
+config = context.config
+
+# Interpret the config file for Python logging.
+# This line sets up loggers basically.
+if config.config_file_name is not None:
+    fileConfig(config.config_file_name)
+
+
+database_url = os.getenv("IOT_DATABASE_URL")
+if database_url is not None:
+    config.set_main_option("sqlalchemy.url", database_url)
+
+# add your model's MetaData object here
+# for 'autogenerate' support
+# from myapp import mymodel
+# target_metadata = mymodel.Base.metadata
+from models.base import Base
+
+target_metadata = Base.metadata
+
+# other values from the config, defined by the needs of env.py,
+# can be acquired:
+# my_important_option = config.get_main_option("my_important_option")
+# ... etc.
+
+
+def run_migrations_offline() -> None:
+    """Run migrations in 'offline' mode.
+
+    This configures the context with just a URL
+    and not an Engine, though an Engine is acceptable
+    here as well.  By skipping the Engine creation
+    we don't even need a DBAPI to be available.
+
+    Calls to context.execute() here emit the given string to the
+    script output.
+
+    """
+    url = config.get_main_option("sqlalchemy.url")
+    context.configure(
+        url=url,
+        target_metadata=target_metadata,
+        literal_binds=True,
+        dialect_opts={"paramstyle": "named"},
+    )
+
+    with context.begin_transaction():
+        context.run_migrations()
+
+
+def run_migrations_online() -> None:
+    """Run migrations in 'online' mode.
+
+    In this scenario we need to create an Engine
+    and associate a connection with the context.
+
+    """
+    connectable = engine_from_config(
+        config.get_section(config.config_ini_section, {}),
+        prefix="sqlalchemy.",
+        poolclass=pool.NullPool,
+    )
+
+    with connectable.connect() as connection:
+        context.configure(connection=connection, target_metadata=target_metadata)
+
+        with context.begin_transaction():
+            context.run_migrations()
+
+
+if context.is_offline_mode():
+    run_migrations_offline()
+else:
+    run_migrations_online()

ngits_iot_db-0.1.0/alembic/script.py.mako

@@ -0,0 +1,26 @@
+"""${message}
+
+Revision ID: ${up_revision}
+Revises: ${down_revision | comma,n}
+Create Date: ${create_date}
+
+"""
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+${imports if imports else ""}
+
+# revision identifiers, used by Alembic.
+revision: str = ${repr(up_revision)}
+down_revision: Union[str, None] = ${repr(down_revision)}
+branch_labels: Union[str, Sequence[str], None] = ${repr(branch_labels)}
+depends_on: Union[str, Sequence[str], None] = ${repr(depends_on)}
+
+
+def upgrade() -> None:
+    ${upgrades if upgrades else "pass"}
+
+
+def downgrade() -> None:
+    ${downgrades if downgrades else "pass"}

ngits_iot_db-0.1.0/alembic/versions/78371c256762_init_schema.py

@@ -0,0 +1,194 @@
+"""Init schema
+
+Revision ID: 78371c256762
+Revises: 
+Create Date: 2024-08-21 00:35:06.960907
+
+"""
+
+from typing import Sequence, Union
+
+import sqlalchemy as sa
+from sqlalchemy.dialects import postgresql
+
+from alembic import op
+
+# revision identifiers, used by Alembic.
+revision: str = "78371c256762"
+down_revision: Union[str, None] = None
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.create_table(
+        "measurement_electricity",
+        sa.Column("measured_ts", sa.DateTime(), nullable=True),
+        sa.Column("value", sa.Float(), nullable=True),
+        sa.Column("id", sa.Integer(), nullable=False),
+        sa.Column("tenant", sa.UUID(), nullable=True),
+        sa.Column("external_id", sa.Text(), nullable=True),
+        sa.Column("meter_id", sa.Text(), nullable=True),
+        sa.Column("created_ts", sa.DateTime(), nullable=True),
+        sa.Column("updated_ts", sa.DateTime(), nullable=True),
+        sa.PrimaryKeyConstraint("id"),
+        sa.UniqueConstraint("tenant", "meter_id", "measured_ts"),
+    )
+    op.create_table(
+        "measurement_heat",
+        sa.Column("measured_ts", sa.DateTime(), nullable=True),
+        sa.Column("value", sa.Float(), nullable=True),
+        sa.Column("id", sa.Integer(), nullable=False),
+        sa.Column("tenant", sa.UUID(), nullable=True),
+        sa.Column("external_id", sa.Text(), nullable=True),
+        sa.Column("meter_id", sa.Text(), nullable=True),
+        sa.Column("created_ts", sa.DateTime(), nullable=True),
+        sa.Column("updated_ts", sa.DateTime(), nullable=True),
+        sa.PrimaryKeyConstraint("id"),
+        sa.UniqueConstraint("tenant", "meter_id", "measured_ts"),
+    )
+    op.create_table(
+        "measurement_raw",
+        sa.Column("data", postgresql.JSON(astext_type=sa.Text()), nullable=True),
+        sa.Column("type", sa.SmallInteger(), nullable=True),
+        sa.Column("is_processed", sa.Boolean(), nullable=True),
+        sa.Column("id", sa.Integer(), nullable=False),
+        sa.Column("tenant", sa.UUID(), nullable=True),
+        sa.Column("external_id", sa.Text(), nullable=True),
+        sa.Column("meter_id", sa.Text(), nullable=True),
+        sa.Column("created_ts", sa.DateTime(), nullable=True),
+        sa.Column("updated_ts", sa.DateTime(), nullable=True),
+        sa.PrimaryKeyConstraint("id"),
+    )
+    op.create_table(
+        "measurement_water",
+        sa.Column("measured_ts", sa.DateTime(), nullable=True),
+        sa.Column("value", sa.Float(), nullable=True),
+        sa.Column("id", sa.Integer(), nullable=False),
+        sa.Column("tenant", sa.UUID(), nullable=True),
+        sa.Column("external_id", sa.Text(), nullable=True),
+        sa.Column("meter_id", sa.Text(), nullable=True),
+        sa.Column("created_ts", sa.DateTime(), nullable=True),
+        sa.Column("updated_ts", sa.DateTime(), nullable=True),
+        sa.PrimaryKeyConstraint("id"),
+        sa.UniqueConstraint("tenant", "meter_id", "measured_ts"),
+    )
+    op.create_table(
+        "usage_electricity_daily",
+        sa.Column("start_ts", sa.DateTime(), nullable=True),
+        sa.Column("end_ts", sa.DateTime(), nullable=True),
+        sa.Column("consumption_ts", sa.DateTime(), nullable=True),
+        sa.Column("start_value", sa.Float(), nullable=True),
+        sa.Column("end_value", sa.Float(), nullable=True),
+        sa.Column("consumption", sa.Float(), nullable=True),
+        sa.Column("id", sa.Integer(), nullable=False),
+        sa.Column("tenant", sa.UUID(), nullable=True),
+        sa.Column("external_id", sa.Text(), nullable=True),
+        sa.Column("meter_id", sa.Text(), nullable=True),
+        sa.Column("created_ts", sa.DateTime(), nullable=True),
+        sa.Column("updated_ts", sa.DateTime(), nullable=True),
+        sa.PrimaryKeyConstraint("id"),
+        sa.UniqueConstraint("tenant", "meter_id", "consumption_ts"),
+    )
+    op.create_table(
+        "usage_electricity_monthly",
+        sa.Column("start_ts", sa.DateTime(), nullable=True),
+        sa.Column("end_ts", sa.DateTime(), nullable=True),
+        sa.Column("consumption_ts", sa.DateTime(), nullable=True),
+        sa.Column("start_value", sa.Float(), nullable=True),
+        sa.Column("end_value", sa.Float(), nullable=True),
+        sa.Column("consumption", sa.Float(), nullable=True),
+        sa.Column("id", sa.Integer(), nullable=False),
+        sa.Column("tenant", sa.UUID(), nullable=True),
+        sa.Column("external_id", sa.Text(), nullable=True),
+        sa.Column("meter_id", sa.Text(), nullable=True),
+        sa.Column("created_ts", sa.DateTime(), nullable=True),
+        sa.Column("updated_ts", sa.DateTime(), nullable=True),
+        sa.PrimaryKeyConstraint("id"),
+        sa.UniqueConstraint("tenant", "meter_id", "consumption_ts"),
+    )
+    op.create_table(
+        "usage_heat_daily",
+        sa.Column("start_ts", sa.DateTime(), nullable=True),
+        sa.Column("end_ts", sa.DateTime(), nullable=True),
+        sa.Column("consumption_ts", sa.DateTime(), nullable=True),
+        sa.Column("start_value", sa.Float(), nullable=True),
+        sa.Column("end_value", sa.Float(), nullable=True),
+        sa.Column("consumption", sa.Float(), nullable=True),
+        sa.Column("id", sa.Integer(), nullable=False),
+        sa.Column("tenant", sa.UUID(), nullable=True),
+        sa.Column("external_id", sa.Text(), nullable=True),
+        sa.Column("meter_id", sa.Text(), nullable=True),
+        sa.Column("created_ts", sa.DateTime(), nullable=True),
+        sa.Column("updated_ts", sa.DateTime(), nullable=True),
+        sa.PrimaryKeyConstraint("id"),
+        sa.UniqueConstraint("tenant", "meter_id", "consumption_ts"),
+    )
+    op.create_table(
+        "usage_heat_monthly",
+        sa.Column("start_ts", sa.DateTime(), nullable=True),
+        sa.Column("end_ts", sa.DateTime(), nullable=True),
+        sa.Column("consumption_ts", sa.DateTime(), nullable=True),
+        sa.Column("start_value", sa.Float(), nullable=True),
+        sa.Column("end_value", sa.Float(), nullable=True),
+        sa.Column("consumption", sa.Float(), nullable=True),
+        sa.Column("id", sa.Integer(), nullable=False),
+        sa.Column("tenant", sa.UUID(), nullable=True),
+        sa.Column("external_id", sa.Text(), nullable=True),
+        sa.Column("meter_id", sa.Text(), nullable=True),
+        sa.Column("created_ts", sa.DateTime(), nullable=True),
+        sa.Column("updated_ts", sa.DateTime(), nullable=True),
+        sa.PrimaryKeyConstraint("id"),
+        sa.UniqueConstraint("tenant", "meter_id", "consumption_ts"),
+    )
+    op.create_table(
+        "usage_water_daily",
+        sa.Column("start_ts", sa.DateTime(), nullable=True),
+        sa.Column("end_ts", sa.DateTime(), nullable=True),
+        sa.Column("consumption_ts", sa.DateTime(), nullable=True),
+        sa.Column("start_value", sa.Float(), nullable=True),
+        sa.Column("end_value", sa.Float(), nullable=True),
+        sa.Column("consumption", sa.Float(), nullable=True),
+        sa.Column("id", sa.Integer(), nullable=False),
+        sa.Column("tenant", sa.UUID(), nullable=True),
+        sa.Column("external_id", sa.Text(), nullable=True),
+        sa.Column("meter_id", sa.Text(), nullable=True),
+        sa.Column("created_ts", sa.DateTime(), nullable=True),
+        sa.Column("updated_ts", sa.DateTime(), nullable=True),
+        sa.PrimaryKeyConstraint("id"),
+        sa.UniqueConstraint("tenant", "meter_id", "consumption_ts"),
+    )
+    op.create_table(
+        "usage_water_monthly",
+        sa.Column("start_ts", sa.DateTime(), nullable=True),
+        sa.Column("end_ts", sa.DateTime(), nullable=True),
+        sa.Column("consumption_ts", sa.DateTime(), nullable=True),
+        sa.Column("start_value", sa.Float(), nullable=True),
+        sa.Column("end_value", sa.Float(), nullable=True),
+        sa.Column("consumption", sa.Float(), nullable=True),
+        sa.Column("id", sa.Integer(), nullable=False),
+        sa.Column("tenant", sa.UUID(), nullable=True),
+        sa.Column("external_id", sa.Text(), nullable=True),
+        sa.Column("meter_id", sa.Text(), nullable=True),
+        sa.Column("created_ts", sa.DateTime(), nullable=True),
+        sa.Column("updated_ts", sa.DateTime(), nullable=True),
+        sa.PrimaryKeyConstraint("id"),
+        sa.UniqueConstraint("tenant", "meter_id", "consumption_ts"),
+    )
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_table("usage_water_monthly")
+    op.drop_table("usage_water_daily")
+    op.drop_table("usage_heat_monthly")
+    op.drop_table("usage_heat_daily")
+    op.drop_table("usage_electricity_monthly")
+    op.drop_table("usage_electricity_daily")
+    op.drop_table("measurement_water")
+    op.drop_table("measurement_raw")
+    op.drop_table("measurement_heat")
+    op.drop_table("measurement_electricity")
+    # ### end Alembic commands ###

ngits_iot_db-0.1.0/alembic/versions/fc391d505594_add_temperature.py

@@ -0,0 +1,43 @@
+"""Add temperature
+
+Revision ID: fc391d505594
+Revises: 78371c256762
+Create Date: 2025-06-15 20:03:15.039980
+
+"""
+
+from typing import Sequence, Union
+
+import sqlalchemy as sa
+
+from alembic import op
+
+# revision identifiers, used by Alembic.
+revision: str = "fc391d505594"
+down_revision: Union[str, None] = "78371c256762"
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.create_table(
+        "measurement_temperature",
+        sa.Column("measured_ts", sa.DateTime(), nullable=True),
+        sa.Column("value", sa.Float(), nullable=True),
+        sa.Column("id", sa.Integer(), nullable=False),
+        sa.Column("tenant", sa.UUID(), nullable=True),
+        sa.Column("external_id", sa.Text(), nullable=True),
+        sa.Column("meter_id", sa.Text(), nullable=True),
+        sa.Column("created_ts", sa.DateTime(), nullable=True),
+        sa.Column("updated_ts", sa.DateTime(), nullable=True),
+        sa.PrimaryKeyConstraint("id"),
+        sa.UniqueConstraint("tenant", "meter_id", "measured_ts"),
+    )
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_table("measurement_temperature")
+    # ### end Alembic commands ###

ngits_iot_db-0.1.0/alembic.ini

@@ -0,0 +1,116 @@
+# A generic, single database configuration.
+
+[alembic]
+# path to migration scripts
+# Use forward slashes (/) also on windows to provide an os agnostic path
+script_location = alembic
+
+# template used to generate migration file names; The default value is %%(rev)s_%%(slug)s
+# Uncomment the line below if you want the files to be prepended with date and time
+# see https://alembic.sqlalchemy.org/en/latest/tutorial.html#editing-the-ini-file
+# for all available tokens
+# file_template = %%(year)d_%%(month).2d_%%(day).2d_%%(hour).2d%%(minute).2d-%%(rev)s_%%(slug)s
+
+# sys.path path, will be prepended to sys.path if present.
+# defaults to the current working directory.
+prepend_sys_path = .
+
+# timezone to use when rendering the date within the migration file
+# as well as the filename.
+# If specified, requires the python>=3.9 or backports.zoneinfo library.
+# Any required deps can installed by adding `alembic[tz]` to the pip requirements
+# string value is passed to ZoneInfo()
+# leave blank for localtime
+# timezone =
+
+# max length of characters to apply to the "slug" field
+# truncate_slug_length = 40
+
+# set to 'true' to run the environment during
+# the 'revision' command, regardless of autogenerate
+# revision_environment = false
+
+# set to 'true' to allow .pyc and .pyo files without
+# a source .py file to be detected as revisions in the
+# versions/ directory
+# sourceless = false
+
+# version location specification; This defaults
+# to alembic/versions.  When using multiple version
+# directories, initial revisions must be specified with --version-path.
+# The path separator used here should be the separator specified by "version_path_separator" below.
+# version_locations = %(here)s/bar:%(here)s/bat:alembic/versions
+
+# version path separator; As mentioned above, this is the character used to split
+# version_locations. The default within new alembic.ini files is "os", which uses os.pathsep.
+# If this key is omitted entirely, it falls back to the legacy behavior of splitting on spaces and/or commas.
+# Valid values for version_path_separator are:
+#
+# version_path_separator = :
+# version_path_separator = ;
+# version_path_separator = space
+version_path_separator = os  # Use os.pathsep. Default configuration used for new projects.
+
+# set to 'true' to search source files recursively
+# in each "version_locations" directory
+# new in Alembic version 1.10
+# recursive_version_locations = false
+
+# the output encoding used when revision files
+# are written from script.py.mako
+# output_encoding = utf-8
+
+sqlalchemy.url = postgresql://postgres:postgres@localhost:5433/iot_local
+
+
+[post_write_hooks]
+# post_write_hooks defines scripts or Python functions that are run
+# on newly generated revision scripts.  See the documentation for further
+# detail and examples
+
+# format using "black" - use the console_scripts runner, against the "black" entrypoint
+# hooks = black
+# black.type = console_scripts
+# black.entrypoint = black
+# black.options = -l 79 REVISION_SCRIPT_FILENAME
+
+# lint with attempts to fix using "ruff" - use the exec runner, execute a binary
+# hooks = ruff
+# ruff.type = exec
+# ruff.executable = %(here)s/.venv/bin/ruff
+# ruff.options = --fix REVISION_SCRIPT_FILENAME
+
+# Logging configuration
+[loggers]
+keys = root,sqlalchemy,alembic
+
+[handlers]
+keys = console
+
+[formatters]
+keys = generic
+
+[logger_root]
+level = WARN
+handlers = console
+qualname =
+
+[logger_sqlalchemy]
+level = WARN
+handlers =
+qualname = sqlalchemy.engine
+
+[logger_alembic]
+level = INFO
+handlers =
+qualname = alembic
+
+[handler_console]
+class = StreamHandler
+args = (sys.stderr,)
+level = NOTSET
+formatter = generic
+
+[formatter_generic]
+format = %(levelname)-5.5s [%(name)s] %(message)s
+datefmt = %H:%M:%S

ngits_iot_db-0.1.0/models/__init__.py

@@ -0,0 +1,77 @@
+"""IoT Database Models - SQLAlchemy models for IoT sensor data storage."""
+
+__version__ = "0.1.0"
+__author__ = "NG IT Services Sp. z o.o."
+__email__ = "biuro@ngits.pl"
+
+from .base import MeasurementBase, Types, Units
+from .electricity import (
+    DailyElectricityUsage,
+    ElectricityMeasurement,
+    MonthlyElectricityUsage,
+)
+from .heat import DailyHeatUsage, HeatMeasurement, MonthlyHeatUsage
+from .raw import RawMeasurement
+from .temperature import TemperatureMeasurement
+from .water import DailyWaterUsage, MonthlyWaterUsage, WaterMeasurement
+
+__all__ = [
+    "DailyElectricityUsage",
+    "DailyHeatUsage",
+    "DailyWaterUsage",
+    "ElectricityMeasurement",
+    "HeatMeasurement",
+    "map_daily_usage_class",
+    "map_measurement_class",
+    "map_monthly_usage_class",
+    "MonthlyElectricityUsage",
+    "MonthlyHeatUsage",
+    "MonthlyWaterUsage",
+    "RawMeasurement",
+    "TemperatureMeasurement",
+    "Types",
+    "Units",
+    "WaterMeasurement",
+]
+
+
+def map_measurement_class(type_: Types):
+    match type_:
+        case Types.electricity:
+            return ElectricityMeasurement
+        case Types.water:
+            return WaterMeasurement
+        case Types.heat:
+            return HeatMeasurement
+        case Types.temperature:
+            return TemperatureMeasurement
+        case _:
+            raise NotImplementedError
+
+
+def map_daily_usage_class(type_: Types):
+    match type_:
+        case Types.electricity:
+            return DailyElectricityUsage
+        case Types.water:
+            return DailyWaterUsage
+        case Types.heat:
+            return DailyHeatUsage
+        case Types.temperature:
+            return None
+        case _:
+            raise NotImplementedError
+
+
+def map_monthly_usage_class(type_: Types):
+    match type_:
+        case Types.electricity:
+            return MonthlyElectricityUsage
+        case Types.water:
+            return MonthlyWaterUsage
+        case Types.heat:
+            return MonthlyHeatUsage
+        case Types.temperature:
+            return None
+        case _:
+            raise NotImplementedError

ngits_iot_db-0.1.0/models/base.py

@@ -0,0 +1,57 @@
+from datetime import datetime
+from enum import Enum as PyEnum
+from enum import IntEnum
+
+from sqlalchemy import Column, DateTime, Float, Integer, Text
+from sqlalchemy.dialects.postgresql import UUID
+from sqlalchemy.ext.declarative import declarative_base
+
+Base = declarative_base()
+
+
+class Units(PyEnum):
+    m3 = "m3"
+    kwh = "kwh"
+    degc = "degc"
+
+
+class Types(IntEnum):
+    water = 1
+    electricity = 2
+    heat = 3
+    temperature = 4
+
+
+class TimeSign(Base):
+    __abstract__ = True
+
+    created_ts = Column(DateTime, default=datetime.utcnow)
+    updated_ts = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
+
+
+class IdentityBase(Base):
+    __abstract__ = True
+
+    id = Column(Integer, primary_key=True)
+    tenant = Column(UUID(as_uuid=True))
+    external_id = Column(Text, nullable=True)
+    meter_id = Column(Text)
+
+
+class MeasurementBase(IdentityBase, TimeSign, Base):
+    __abstract__ = True
+
+    measured_ts = Column(DateTime)
+    value = Column(Float)
+
+
+class UsageBase(IdentityBase, TimeSign, Base):
+    __abstract__ = True
+
+    start_ts = Column(DateTime)
+    end_ts = Column(DateTime)
+    consumption_ts = Column(DateTime)
+
+    start_value = Column(Float)
+    end_value = Column(Float)
+    consumption = Column(Float)

ngits_iot_db-0.1.0/models/electricity.py

@@ -0,0 +1,38 @@
+from sqlalchemy import UniqueConstraint
+
+from .base import MeasurementBase, UsageBase
+
+
+class ElectricityMeasurement(MeasurementBase):
+    __tablename__ = "measurement_electricity"
+    __table_args__ = (
+        UniqueConstraint(
+            "tenant",
+            "meter_id",
+            "measured_ts",
+        ),
+    )
+
+    RAW_VALUE_ATTR_NAME = "total_energy_consumption_kwh"
+
+
+class DailyElectricityUsage(UsageBase):
+    __tablename__ = "usage_electricity_daily"
+    __table_args__ = (
+        UniqueConstraint(
+            "tenant",
+            "meter_id",
+            "consumption_ts",
+        ),
+    )
+
+
+class MonthlyElectricityUsage(UsageBase):
+    __tablename__ = "usage_electricity_monthly"
+    __table_args__ = (
+        UniqueConstraint(
+            "tenant",
+            "meter_id",
+            "consumption_ts",
+        ),
+    )

ngits_iot_db-0.1.0/models/heat.py

@@ -0,0 +1,37 @@
+from sqlalchemy import UniqueConstraint
+
+from .base import MeasurementBase, UsageBase
+
+
+class HeatMeasurement(MeasurementBase):
+    __tablename__ = "measurement_heat"
+    __table_args__ = (
+        UniqueConstraint(
+            "tenant",
+            "meter_id",
+            "measured_ts",
+        ),
+    )
+    RAW_VALUE_ATTR_NAME = "total_energy_consumption_kwh"
+
+
+class DailyHeatUsage(UsageBase):
+    __tablename__ = "usage_heat_daily"
+    __table_args__ = (
+        UniqueConstraint(
+            "tenant",
+            "meter_id",
+            "consumption_ts",
+        ),
+    )
+
+
+class MonthlyHeatUsage(UsageBase):
+    __tablename__ = "usage_heat_monthly"
+    __table_args__ = (
+        UniqueConstraint(
+            "tenant",
+            "meter_id",
+            "consumption_ts",
+        ),
+    )

ngits_iot_db-0.1.0/models/raw.py

@@ -0,0 +1,20 @@
+from sqlalchemy import Boolean, Column, SmallInteger
+from sqlalchemy.dialects.postgresql import JSON
+
+from .base import Base, IdentityBase, TimeSign, Types
+
+
+class RawMeasurement(IdentityBase, TimeSign, Base):
+    __tablename__ = "measurement_raw"
+
+    data = Column(JSON)
+    type = Column(SmallInteger)
+    is_processed = Column(Boolean, default=False)
+
+    @property
+    def type_enum(self):
+        return Types(self.type)
+
+    @type_enum.setter
+    def type_enum(self, type_enum):
+        self.type = type_enum.value

ngits_iot_db-0.1.0/models/temperature.py

@@ -0,0 +1,15 @@
+from sqlalchemy import UniqueConstraint
+
+from .base import MeasurementBase, UsageBase
+
+
+class TemperatureMeasurement(MeasurementBase):
+    __tablename__ = "measurement_temperature"
+    __table_args__ = (
+        UniqueConstraint(
+            "tenant",
+            "meter_id",
+            "measured_ts",
+        ),
+    )
+    RAW_VALUE_ATTR_NAME = "total_degc"

ngits_iot_db-0.1.0/models/water.py

@@ -0,0 +1,37 @@
+from sqlalchemy import UniqueConstraint
+
+from .base import MeasurementBase, UsageBase
+
+
+class WaterMeasurement(MeasurementBase):
+    __tablename__ = "measurement_water"
+    __table_args__ = (
+        UniqueConstraint(
+            "tenant",
+            "meter_id",
+            "measured_ts",
+        ),
+    )
+    RAW_VALUE_ATTR_NAME = "total_m3"
+
+
+class DailyWaterUsage(UsageBase):
+    __tablename__ = "usage_water_daily"
+    __table_args__ = (
+        UniqueConstraint(
+            "tenant",
+            "meter_id",
+            "consumption_ts",
+        ),
+    )
+
+
+class MonthlyWaterUsage(UsageBase):
+    __tablename__ = "usage_water_monthly"
+    __table_args__ = (
+        UniqueConstraint(
+            "tenant",
+            "meter_id",
+            "consumption_ts",
+        ),
+    )

ngits_iot_db-0.1.0/ngits_iot_db.egg-info/PKG-INFO

@@ -0,0 +1,67 @@
+Metadata-Version: 2.4
+Name: ngits-iot-db
+Version: 0.1.0
+Summary: SQLAlchemy models for IoT sensor data storage with Alembic migrations
+Author-email: "NG IT Services Sp. z o. o." <biuro@ngits.pl>
+Keywords: iot,sqlalchemy,database,models,sensors
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: BSD License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Topic :: Database
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: SQLAlchemy>=2.0.0
+Requires-Dist: alembic>=1.13.0
+Requires-Dist: psycopg2-binary>=2.9.0
+Provides-Extra: dev
+Requires-Dist: black>=25.1.0; extra == "dev"
+Requires-Dist: isort>=5.13.0; extra == "dev"
+Requires-Dist: mypy>=1.0.0; extra == "dev"
+Requires-Dist: pytest>=7.0.0; extra == "dev"
+Requires-Dist: pytest-cov>=4.0.0; extra == "dev"
+Dynamic: license-file
+
+database uri
+============
+
+You can set database uri by setting `IOT_DATABASE_URL` variable.
+
+```shell
+export IOT_DATABASE_URL=postgresql://postgres:postgres@localhost:5432/iot_local
+```
+
+
+make migrations
+===============
+
+```shell
+alembic revision --autogenerate -m "Migration message."
+```
+
+upgrade
+=======
+
+```shell
+# upgrade all
+alembic upgrade head
+
+# upgrade all to specific migration
+alembic upgrade <revision_id>
+```
+
+downgrade
+=========
+
+```shell
+# downgrade all
+alembic downgrade base
+
+# downgrade all to specific migration
+alembic downgrade <revision_id>
+```

ngits_iot_db-0.1.0/ngits_iot_db.egg-info/SOURCES.txt

@@ -0,0 +1,23 @@
+LICENSE
+MANIFEST.in
+README.md
+alembic.ini
+pyproject.toml
+requirements.txt
+alembic/env.py
+alembic/script.py.mako
+alembic/versions/78371c256762_init_schema.py
+alembic/versions/fc391d505594_add_temperature.py
+models/__init__.py
+models/base.py
+models/electricity.py
+models/heat.py
+models/py.typed
+models/raw.py
+models/temperature.py
+models/water.py
+ngits_iot_db.egg-info/PKG-INFO
+ngits_iot_db.egg-info/SOURCES.txt
+ngits_iot_db.egg-info/dependency_links.txt
+ngits_iot_db.egg-info/requires.txt
+ngits_iot_db.egg-info/top_level.txt

ngits_iot_db-0.1.0/ngits_iot_db.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

ngits_iot_db-0.1.0/ngits_iot_db.egg-info/requires.txt

@@ -0,0 +1,10 @@
+SQLAlchemy>=2.0.0
+alembic>=1.13.0
+psycopg2-binary>=2.9.0
+
+[dev]
+black>=25.1.0
+isort>=5.13.0
+mypy>=1.0.0
+pytest>=7.0.0
+pytest-cov>=4.0.0

ngits_iot_db-0.1.0/ngits_iot_db.egg-info/top_level.txt

@@ -0,0 +1 @@
+models

ngits_iot_db-0.1.0/pyproject.toml

@@ -0,0 +1,95 @@
+[build-system]
+requires = ["setuptools>=61.0", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "ngits-iot-db"
+version = "0.1.0"
+description = "SQLAlchemy models for IoT sensor data storage with Alembic migrations"
+readme = "README.md"
+requires-python = ">=3.10"
+authors = [
+    {name = "NG IT Services Sp. z o. o.", email = "biuro@ngits.pl"}
+]
+keywords = ["iot", "sqlalchemy", "database", "models", "sensors"]
+classifiers = [
+    "Development Status :: 3 - Alpha",
+    "Intended Audience :: Developers",
+    "License :: OSI Approved :: BSD License",
+    "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3.10",
+    "Programming Language :: Python :: 3.11",
+    "Programming Language :: Python :: 3.12",
+    "Topic :: Database",
+    "Topic :: Software Development :: Libraries :: Python Modules",
+]
+
+dependencies = [
+    "SQLAlchemy>=2.0.0",
+    "alembic>=1.13.0",
+    "psycopg2-binary>=2.9.0",
+]
+
+[project.optional-dependencies]
+dev = [
+    "black>=25.1.0",
+    "isort>=5.13.0",
+    "mypy>=1.0.0",
+    "pytest>=7.0.0",
+    "pytest-cov>=4.0.0",
+]
+
+[tool.setuptools.packages.find]
+where = ["."]
+include = ["models*"]
+exclude = ["tests*", "alembic*"]
+
+[tool.setuptools.package-data]
+models = ["py.typed"]
+
+[tool.black]
+line-length = 88
+target-version = ['py310']
+include = '\.pyi?$'
+extend-exclude = '''
+/(
+  # directories
+  \.eggs
+  | \.git
+  | \.hg
+  | \.mypy_cache
+  | \.tox
+  | \.venv
+  | _build
+  | buck-out
+  | build
+  | dist
+  | alembic/versions
+)/
+'''
+
+[tool.isort]
+profile = "black"
+multi_line_output = 3
+include_trailing_comma = true
+force_grid_wrap = 0
+use_parentheses = true
+ensure_newline_before_comments = true
+line_length = 88
+skip_glob = ["alembic/versions/*.py"]
+
+[tool.mypy]
+python_version = "3.9"
+warn_return_any = true
+warn_unused_configs = true
+disallow_untyped_defs = true
+disallow_incomplete_defs = true
+check_untyped_defs = true
+disallow_untyped_decorators = true
+no_implicit_optional = true
+warn_redundant_casts = true
+warn_unused_ignores = true
+warn_no_return = true
+warn_unreachable = true
+strict_equality = true
+exclude = ["alembic/"]

ngits_iot_db-0.1.0/requirements.txt

@@ -0,0 +1,14 @@
+alembic==1.13.2
+black==25.1.0
+click==8.1.7
+greenlet==3.2.3
+isort==5.13.2
+Mako==1.3.5
+MarkupSafe==2.1.5
+mypy-extensions==1.0.0
+packaging==25.0
+pathspec==0.12.1
+platformdirs==4.2.2
+psycopg2-binary==2.9.10
+SQLAlchemy==2.0.32
+typing_extensions==4.12.2

ngits_iot_db-0.1.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+