grid-data-models 0.0.0__tar.gz

grid_data_models-0.0.0/.gitignore

@@ -0,0 +1,163 @@
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# C extensions
+*.so
+*.code-workspace
+*.sqlite
+*.ruff_cache/
+
+# Distribution / packaging
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+share/python-wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
+
+# PyInstaller
+#  Usually these files are written by a python script from a template
+#  before PyInstaller builds the exe, so as to inject date/other infos into it.
+*.manifest
+*.spec
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.nox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+*.py,cover
+.hypothesis/
+.pytest_cache/
+cover/
+
+# Translations
+*.mo
+*.pot
+
+# Django stuff:
+*.log
+local_settings.py
+db.sqlite3
+db.sqlite3-journal
+
+# Flask stuff:
+instance/
+.webassets-cache
+
+# Scrapy stuff:
+.scrapy
+
+# Sphinx documentation
+docs/_build/
+
+# PyBuilder
+.pybuilder/
+target/
+
+# Jupyter Notebook
+.ipynb_checkpoints
+
+# IPython
+profile_default/
+ipython_config.py
+
+# pyenv
+#   For a library or package, you might want to ignore these files since the code is
+#   intended to run in multiple environments; otherwise, check them in:
+# .python-version
+
+# pipenv
+#   According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
+#   However, in case of collaboration, if having platform-specific dependencies or dependencies
+#   having no cross-platform support, pipenv may install dependencies that don't work, or not
+#   install all needed dependencies.
+#Pipfile.lock
+
+# poetry
+#   Similar to Pipfile.lock, it is generally recommended to include poetry.lock in version control.
+#   This is especially recommended for binary packages to ensure reproducibility, and is more
+#   commonly ignored for libraries.
+#   https://python-poetry.org/docs/basic-usage/#commit-your-poetrylock-file-to-version-control
+#poetry.lock
+
+# pdm
+#   Similar to Pipfile.lock, it is generally recommended to include pdm.lock in version control.
+#pdm.lock
+#   pdm stores project-wide configurations in .pdm.toml, but it is recommended to not include it
+#   in version control.
+#   https://pdm.fming.dev/#use-with-ide
+.pdm.toml
+
+# PEP 582; used by e.g. github.com/David-OConnor/pyflow and github.com/pdm-project/pdm
+__pypackages__/
+
+# Celery stuff
+celerybeat-schedule
+celerybeat.pid
+
+# SageMath parsed files
+*.sage.py
+
+# Environments
+.env
+.venv
+env/
+venv/
+ENV/
+env.bak/
+venv.bak/
+
+# Spyder project settings
+.spyderproject
+.spyproject
+
+# Rope project settings
+.ropeproject
+
+# mkdocs documentation
+/site
+
+# mypy
+.mypy_cache/
+.dmypy.json
+dmypy.json
+
+# Pyre type checker
+.pyre/
+
+# pytype static type analyzer
+.pytype/
+
+# Cython debug symbols
+cython_debug/
+
+# PyCharm
+#  JetBrains specific template is maintained in a separate JetBrains.gitignore that can
+#  be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore
+#  and can be added to the global gitignore or merged into this file.  For a more nuclear
+#  option (not recommended) you can uncomment the following to ignore the entire idea folder.
+#.idea/

grid_data_models-0.0.0/LICENSE.txt

@@ -0,0 +1,29 @@
+BSD 3-Clause License
+
+Copyright (c) 2024, Alliance for Sustainable Energy, LLC
+All rights reserved.
+
+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.

grid_data_models-0.0.0/PKG-INFO

@@ -0,0 +1,54 @@
+Metadata-Version: 2.3
+Name: grid-data-models
+Version: 0.0.0
+Project-URL: Documentation, https://github.com/NREL-Distribution-Suites/grid-data-models#readme
+Project-URL: Issues, https://github.com/NREL-Distribution-Suites/grid-data-models/issues
+Project-URL: Source, https://github.com/NREL-Distribution-Suites/grid-data-models
+Author-email: Kapil Duwadi <Kapil.Duwadi@nrel.gov>, Aadil Latif <Aadil.Latif@nrel.gov>, Tarek Elgindy <tarek.elgindy@nrel.gov>
+License-Expression: BSD-3-Clause
+License-File: LICENSE.txt
+Classifier: License :: OSI Approved :: BSD License
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3.11
+Requires-Python: >=3.11
+Requires-Dist: infrasys
+Requires-Dist: networkx
+Provides-Extra: dev
+Requires-Dist: pre-commit; extra == 'dev'
+Requires-Dist: pytest; extra == 'dev'
+Requires-Dist: pytest-cov; extra == 'dev'
+Requires-Dist: ruff; extra == 'dev'
+Provides-Extra: doc
+Requires-Dist: myst-parser; extra == 'doc'
+Requires-Dist: sphinx; extra == 'doc'
+Requires-Dist: sphinx-immaterial; extra == 'doc'
+Requires-Dist: sphinxcontrib-mermaid; extra == 'doc'
+Description-Content-Type: text/markdown
+
+# Grid Data Models (GDM)
+
+GDM is a python package containing [pydantic](https://docs.pydantic.dev/latest/) data models for power system assets and datasets. This package is actively being developed at [National Renewable Energy Laboratory (NREL)](https://www.nrel.gov/).
+
+## Why Grid Data Models ?
+
+In an effort to reduce code duplication and provide client packages a standard interface to interact with power system data, a group of 
+research engineers at NREL is working on developing standard data models. Features:
+
+- **Builtin validation layer:** Use of [pydantic](https://docs.pydantic.dev/latest/) in creating data models allows us to check for fields during the time of construction and update.
+- **Timeseries data management:** GDM uses [infrasys](https://github.nrel.gov/CADET/infrastructure_systems) package which enables attaching time series data to fields in the data model. For example, we can attach time series power consumption data to a load profile.
+- **Builtin unit conversion:** GDM leverages [pint](https://pint.readthedocs.io/en/stable/) for unit conversion for power system quantities. For e.g power, voltage, time etc.
+- **JSON serialization/deserializatin:** GDM uses [infrasys](https://github.nrel.gov/CADET/infrastructure_systems) to serialize and deserialize distribution system containing power system components and time series data attached to components.
+
+## How to get started ?
+
+To get started, you can clone and pip install this library from [here](https://github.nrel.gov/CADET/grid-data-models).
+
+
+## Contributors
+
+- **Kapil Duwadi**
+- **Tarek Elgindy**
+- **Aadil Latif**
+- **Pedro Andres Sanchez Perez**
+- **Daniel Thompson**
+- **Jeremy Keen**

grid_data_models-0.0.0/README.md

@@ -0,0 +1,27 @@
+# Grid Data Models (GDM)
+
+GDM is a python package containing [pydantic](https://docs.pydantic.dev/latest/) data models for power system assets and datasets. This package is actively being developed at [National Renewable Energy Laboratory (NREL)](https://www.nrel.gov/).
+
+## Why Grid Data Models ?
+
+In an effort to reduce code duplication and provide client packages a standard interface to interact with power system data, a group of 
+research engineers at NREL is working on developing standard data models. Features:
+
+- **Builtin validation layer:** Use of [pydantic](https://docs.pydantic.dev/latest/) in creating data models allows us to check for fields during the time of construction and update.
+- **Timeseries data management:** GDM uses [infrasys](https://github.nrel.gov/CADET/infrastructure_systems) package which enables attaching time series data to fields in the data model. For example, we can attach time series power consumption data to a load profile.
+- **Builtin unit conversion:** GDM leverages [pint](https://pint.readthedocs.io/en/stable/) for unit conversion for power system quantities. For e.g power, voltage, time etc.
+- **JSON serialization/deserializatin:** GDM uses [infrasys](https://github.nrel.gov/CADET/infrastructure_systems) to serialize and deserialize distribution system containing power system components and time series data attached to components.
+
+## How to get started ?
+
+To get started, you can clone and pip install this library from [here](https://github.nrel.gov/CADET/grid-data-models).
+
+
+## Contributors
+
+- **Kapil Duwadi**
+- **Tarek Elgindy**
+- **Aadil Latif**
+- **Pedro Andres Sanchez Perez**
+- **Daniel Thompson**
+- **Jeremy Keen**

grid_data_models-0.0.0/pyproject.toml

@@ -0,0 +1,81 @@
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[project]
+name = "grid-data-models"
+dynamic = ["version"]
+description = ''
+readme = "README.md"
+requires-python = ">=3.11"
+license = "BSD-3-Clause"
+keywords = []
+authors = [
+  { name = "Kapil Duwadi", email = "Kapil.Duwadi@nrel.gov" },
+  { name = "Aadil Latif", email = "Aadil.Latif@nrel.gov" },
+  { name = "Tarek Elgindy", email = "tarek.elgindy@nrel.gov" },
+]
+classifiers = [
+  "Programming Language :: Python :: 3.11",
+  "License :: OSI Approved :: BSD License",
+  "Operating System :: OS Independent",
+]
+dependencies = ["networkx", "infrasys"]
+
+[project.optional-dependencies]
+dev = ["pre-commit", "pytest", "pytest-cov", "ruff"]
+doc = ["sphinx", "sphinx-immaterial", "myst-parser", "sphinxcontrib-mermaid"]
+
+[project.urls]
+Documentation = "https://github.com/NREL-Distribution-Suites/grid-data-models#readme"
+Issues = "https://github.com/NREL-Distribution-Suites/grid-data-models/issues"
+Source = "https://github.com/NREL-Distribution-Suites/grid-data-models"
+
+[tool.ruff]
+# Exclude a variety of commonly ignored directories.
+exclude = [".git", ".ruff_cache", ".venv", "_build", "build", "dist", "venv"]
+ignore-init-module-imports = true
+line-length = 99
+indent-width = 4
+target-version = "py311"
+
+[tool.hatch.version]
+path = 'src/gdm/version.py'
+
+
+[tool.ruff.lint]
+# Enable Pyflakes (`F`) and a subset of the pycodestyle (`E`)  codes by default.
+select = [
+  "C901", # McCabe complexity
+  "E4",   # Subset of pycodestyle (E)
+  "E7",
+  "E9",
+  "F",    # Pyflakes
+  "W",    # pycodestyle warnings
+]
+ignore = []
+
+# Allow fix for all enabled rules (when `--fix`) is provided.
+fixable = ["ALL"]
+unfixable = []
+
+[tool.ruff.format]
+quote-style = "double"
+indent-style = "space"
+skip-magic-trailing-comma = false
+line-ending = "auto"
+docstring-code-format = true
+docstring-code-line-length = "dynamic"
+
+[tool.hatch.metadata]
+allow-direct-references = true
+
+[tool.ruff.lint.per-file-ignores]
+"__init__.py" = ["E402", "F401"]
+"**/{tests,docs,tools}/*" = ["E402"]
+
+[tool.hatch.build.targets.wheel]
+packages = ["src/gdm"]
+
+[tool.hatch.build.targets.sdist]
+include = ["src/gdm"]

grid_data_models-0.0.0/src/gdm/__init__.py

@@ -0,0 +1,107 @@
+""" Managing imports for this package."""
+
+from gdm.distribution.components.distribution_substation import DistributionSubstation
+from gdm.distribution.components.distribution_component import DistributionComponent
+from gdm.distribution.components.distribution_feeder import DistributionFeeder
+from gdm.distribution.components.distribution_bus import DistributionBus
+from gdm.distribution.components.distribution_vsource import (
+    PhaseVoltageSourceEquipment,
+    VoltageSourceEquipment,
+    DistributionVoltageSource,
+)
+from gdm.distribution.components.distribution_branch import (
+    DistributionBranch,
+    SwitchedDistributionBranch,
+)
+from gdm.distribution.components.sequence_impedance_branch import SequenceImpedanceBranch
+from gdm.distribution.components.matrix_impedance_branch import MatrixImpedanceBranch
+from gdm.distribution.components.geometry_branch import GeometryBranch
+from gdm.distribution.components.distribution_switch import DistributionSwitch
+from gdm.distribution.components.matrix_impedance_switch import MatrixImpedanceSwitch
+from gdm.distribution.components.distribution_fuse import DistributionFuse
+from gdm.distribution.components.matrix_impedance_fuse import MatrixImpedanceFuse
+from gdm.distribution.components.distribution_recloser import DistributionRecloser
+from gdm.distribution.components.matrix_impedance_recloser import MatrixImpedanceRecloser
+from gdm.distribution.components.distribution_load import DistributionLoad
+from gdm.distribution.components.distribution_transformer import (
+    DistributionTransformer,
+)
+from gdm.distribution.components.distribution_regulator import DistributionRegulator
+from gdm.distribution.components.distribution_solar import (
+    DistributionSolar,
+)
+from gdm.distribution.components.distribution_capacitor import DistributionCapacitor
+
+from gdm.distribution.equipment.matrix_impedance_branch_equipment import (
+    MatrixImpedanceBranchEquipment,
+)
+from gdm.distribution.equipment.bare_conductor_equipment import BareConductorEquipment
+from gdm.distribution.equipment.concentric_cable_equipment import ConcentricCableEquipment
+from gdm.distribution.equipment.matrix_impedance_fuse_equipment import MatrixImpedanceFuseEquipment
+from gdm.distribution.equipment.matrix_impedance_recloser_equipment import (
+    MatrixImpedanceRecloserEquipment,
+)
+from gdm.distribution.equipment.matrix_impedance_switch_equipment import (
+    MatrixImpedanceSwitchEquipment,
+)
+from gdm.distribution.equipment.recloser_controller_equipment import RecloserControllerEquipment
+from gdm.distribution.equipment.phase_capacitor_equipment import PhaseCapacitorEquipment
+from gdm.distribution.equipment.capacitor_equipment import CapacitorEquipment
+from gdm.distribution.equipment.geometry_branch_equipment import GeometryBranchEquipment
+from gdm.distribution.equipment.sequence_impedance_branch_equipment import (
+    SequenceImpedanceBranchEquipment,
+)
+from gdm.distribution.equipment.phase_load_equipment import PhaseLoadEquipment
+from gdm.distribution.equipment.load_equipment import LoadEquipment
+
+from gdm.distribution.equipment.distribution_transformer_equipment import (
+    DistributionTransformerEquipment,
+    TapWindingEquipment,
+    WindingEquipment,
+)
+from gdm.distribution.equipment.solar_equipment import SolarEquipment
+from gdm.distribution.equipment.inverter_equipment import InverterEquipment
+
+from gdm.distribution.controllers.distribution_inverter_controller import (
+    InverterController,
+    PowerfactorInverterController,
+    VoltVarInverterController,
+    VoltVarVoltWattInverterController,
+    VoltWattInverterController,
+)
+from gdm.distribution.controllers.distribution_capacitor_controller import (
+    CapacitorController,
+    VoltageCapacitorController,
+    ActivePowerCapacitorController,
+    ReactivePowerCapacitorController,
+    CurrentCapacitorController,
+    DailyTimedCapacitorController,
+)
+from gdm.distribution.controllers.distribution_regulator_controller import (
+    RegulatorController,
+)
+from gdm.distribution.controllers.distribution_switch_controller import (
+    DistributionSwitchController,
+)
+from gdm.distribution.controllers.distribution_recloser_controller import (
+    DistributionRecloserController,
+)
+
+from gdm.distribution.sequence_pair import SequencePair
+from gdm.distribution.limitset import ThermalLimitSet, VoltageLimitSet
+from gdm.distribution.distribution_enum import (
+    Phase,
+    ConnectionType,
+    VoltageTypes,
+    LimitType,
+)
+from gdm.distribution.distribution_system import DistributionSystem
+from gdm.distribution.distribution_graph import build_graph_from_system
+from gdm.distribution.curve import Curve, TimeCurrentCurve
+
+from gdm.transmission.transmission_bus import TransmissionBus
+from gdm.transmission.transmission_component import TransmissionComponent
+from gdm.transmission.transmission_branch import TransmissionBranch
+from gdm.transmission.transmission_capacitor import TransmissionCapacitor
+from gdm.transmission.transmission_load import TransmissionLoad
+from gdm.transmission.transmission_substation import TransmissionSubstation

grid_data_models-0.0.0/src/gdm/bus.py

@@ -0,0 +1,28 @@
+""" Interface for power system bus."""
+
+from typing import Annotated, Optional
+
+from infrasys import Component, Location
+from pydantic import Field
+
+from gdm.quantities import PositiveVoltage
+
+
+class PowerSystemBus(Component):
+    """Interface for power system bus."""
+
+    nominal_voltage: Annotated[
+        PositiveVoltage, Field(..., description="Nominal voltage for this bus.")
+    ]
+    coordinate: Annotated[
+        Optional[Location],
+        Field(None, description="Coordinate for the power system bus."),
+    ]
+
+    @classmethod
+    def example(cls) -> "PowerSystemBus":
+        return PowerSystemBus(
+            name="Bus1",
+            nominal_voltage=PositiveVoltage(400, "volt"),
+            coordinate=Location(x=20.0, y=30.0),
+        )

grid_data_models-0.0.0/src/gdm/capacitor.py

@@ -0,0 +1,43 @@
+""" This module contains interface for power system capacitor."""
+
+# pylint:disable=pointless-statement
+
+from typing import Annotated
+
+from pydantic import Field, NonNegativeInt, PositiveInt, model_validator
+
+from infrasys import Component
+from gdm.quantities import PositiveReactivePower
+
+
+class PowerSystemCapacitor(Component):
+    """Interface for power system capacitor."""
+
+    rated_capacity: Annotated[
+        PositiveReactivePower,
+        Field(..., description="Capacity of this capacitor."),
+    ]
+    num_banks_on: Annotated[
+        NonNegativeInt, Field(..., description="Number of banks currently on.")
+    ]
+    num_banks: Annotated[PositiveInt, Field(1, description="Number of banks in the capacitor.")]
+
+    @model_validator(mode="after")
+    def validate_fields(self) -> "PowerSystemCapacitor":
+        """Custom validator for fields."""
+        if self.num_banks < self.num_banks_on:
+            msg = f"Status {self.num_banks_on} must be less than or equal"
+            f"to number of banks. {self.num_banks}"
+            raise ValueError(msg)
+
+        return self
+
+    @classmethod
+    def example(cls) -> "PowerSystemCapacitor":
+        """Example for power system capacitor."""
+        return PowerSystemCapacitor(
+            name="Phase-Cap-1",
+            rated_capacity=PositiveReactivePower(200, "kvar"),
+            num_banks_on=1,
+            num_banks=1,
+        )

grid_data_models-0.0.0/src/gdm/dataset/cost_model.py

@@ -0,0 +1,49 @@
+"""This module contains cost model interface."""
+
+from datetime import datetime
+from enum import Enum
+from typing import Annotated, Optional
+
+
+from pydantic import (
+    PlainSerializer,
+    PositiveFloat,
+    ConfigDict,
+    model_validator,
+)
+from infrasys import Component
+
+
+class OperatingUnitEnum(str, Enum):
+    """Interface for operating unit enumeration."""
+
+    KWH = "kWh"
+    HOUR = "hr"
+    KW = "kW"
+    KVA = "kVA"
+
+
+class CostModel(Component):
+    """Interface for base cost model."""
+
+    purchase_date: Annotated[datetime, PlainSerializer(lambda x: x.strftime("%Y-%m-%d %H:%M:%S"))]
+    capital_dollars: PositiveFloat
+    operating_dollars: Optional[PositiveFloat] = None
+    operating_unit: Optional[OperatingUnitEnum] = None
+    labor_dollars: Optional[PositiveFloat] = None
+    name: str = ""
+    notes: Optional[str] = None
+    location: Optional[str] = None
+    model_config = ConfigDict(arbitrary_types_allowed=True, use_enum_values=False)
+
+    @model_validator(mode="after")
+    def validate_fields(self) -> "CostModel":
+        """Custom validator for cost fields."""
+        if self.operating_dollars is not None and self.operating_unit is None:
+            msg = "Operating unit can not be null when operating dollar is specified"
+            raise ValueError(msg)
+
+    @classmethod
+    def example(cls) -> "CostModel":
+        """Example for cost model."""
+        return CostModel(purchase_date=datetime.utcnow(), capital_dollars=234.45)

grid_data_models-0.0.0/src/gdm/dataset/dataset_system.py

@@ -0,0 +1,76 @@
+"""This module contains dataset store."""
+
+from typing import Any
+from uuid import UUID
+
+from infrasys import System, Component
+
+from gdm.exceptions import GDMIncompatibleInstanceError
+from gdm.dataset.cost_model import CostModel
+import gdm
+
+
+class DatasetSystem(System):
+    """Class interface for dataset system."""
+
+    def __init__(self, *args, catalog_cost_mapping: dict[str, list[str]] | None = None, **kwargs):
+        super().__init__(*args, **kwargs)
+        self.data_format_version = gdm.distribution.__version__
+        self.catalog_cost_mapping = catalog_cost_mapping or {}
+
+    def serialize_system_attributes(self) -> dict[str, Any]:
+        """Method to serialize system attributes."""
+        return {"catalog_cost_mapping": self.catalog_cost_mapping}
+
+    def deserialize_system_attributes(self, data: dict[str, Any]) -> None:
+        """Method to deserialize system attributes."""
+        self.catalog_cost_mapping = data["catalog_cost_mapping"]
+
+    def add_cost(self, catalog: Component, cost: CostModel):
+        """Method to add cost to catalog."""
+
+        if isinstance(catalog, CostModel):
+            msg = f"Catalog can not be an instance of cost model {catalog=}"
+            raise GDMIncompatibleInstanceError(msg)
+
+        self.add_component(cost)
+        cost = self.get_component(CostModel, name=cost.name)
+        catalog_uuid = str(catalog.uuid)
+        cost_uuid = str(cost.uuid)
+
+        if catalog_uuid not in self.catalog_cost_mapping:
+            self.catalog_cost_mapping[catalog_uuid] = []
+
+        if cost_uuid not in self.catalog_cost_mapping[catalog_uuid]:
+            self.catalog_cost_mapping[catalog_uuid].append(cost_uuid)
+
+    def add_costs(self, catalog: Component, costs: list[CostModel]):
+        """Method to add cost to catalog."""
+        for cost in costs:
+            self.add_cost(catalog, cost)
+
+    def get_costs(self, catalog: Component) -> list[CostModel]:
+        """Get costs for a catalog."""
+
+        catalog_uuid = str(catalog.uuid)
+        if catalog_uuid not in self.catalog_cost_mapping:
+            return []
+
+        return [
+            self.get_component_by_uuid(UUID(cost_uuid))
+            for cost_uuid in self.catalog_cost_mapping[catalog_uuid]
+        ]
+
+    def remove_cost(self, catalog: Component, cost: CostModel):
+        """Remove cost from the catalog."""
+
+        self.remove_component(cost)
+
+        catalog_uuid = str(catalog.uuid)
+        cost_uuid = str(cost.uuid)
+
+        if (
+            str(catalog_uuid) in self.catalog_cost_mapping
+            and cost_uuid in self.catalog_cost_mapping[catalog_uuid]
+        ):
+            self.catalog_cost_mapping[catalog_uuid].remove(cost_uuid)

grid_data_models-0.0.0/src/gdm/distribution/__init__.py

@@ -0,0 +1,3 @@
+""" This module stores dataformat version. """
+
+__version__ = "1.0.0"