zepben.ewb 1.0.0b1__py3-none-any.whl

zepben/ewb/database/sqlite/network/network_database_reader.py

@@ -0,0 +1,173 @@
+#  Copyright 2024 Zeppelin Bend Pty Ltd
+#  This Source Code Form is subject to the terms of the Mozilla Public
+#  License, v. 2.0. If a copy of the MPL was not distributed with this
+#  file, You can obtain one at https://mozilla.org/MPL/2.0/.
+
+__all__ = ["NetworkDatabaseReader"]
+
+from collections import Counter
+from sqlite3 import Connection
+from typing import List
+from typing import Optional
+
+from zepben.ewb.database.sqlite.common.base_database_reader import BaseDatabaseReader
+from zepben.ewb.database.sqlite.common.metadata_collection_reader import MetadataCollectionReader
+from zepben.ewb.database.sqlite.network.network_database_tables import NetworkDatabaseTables
+from zepben.ewb.database.sqlite.network.network_service_reader import NetworkServiceReader
+from zepben.ewb.database.sqlite.tables.table_version import TableVersion
+from zepben.ewb.model.cim.iec61970.base.core.equipment import Equipment
+from zepben.ewb.model.cim.iec61970.base.core.feeder import Feeder
+from zepben.ewb.model.cim.iec61970.base.wires.energy_source import EnergySource
+from zepben.ewb.services.network.network_service import NetworkService, connected_equipment
+from zepben.ewb.services.network.tracing.feeder.assign_to_feeders import AssignToFeeders
+from zepben.ewb.services.network.tracing.feeder.assign_to_lv_feeders import AssignToLvFeeders
+from zepben.ewb.services.network.tracing.feeder.set_direction import SetDirection
+from zepben.ewb.services.network.tracing.networktrace.operators.network_state_operators import NetworkStateOperators
+from zepben.ewb.services.network.tracing.networktrace.tracing import Tracing
+from zepben.ewb.services.network.tracing.phases.phase_inferrer import PhaseInferrer
+from zepben.ewb.services.network.tracing.phases.set_phases import SetPhases
+
+
+class NetworkDatabaseReader(BaseDatabaseReader):
+    """
+    A class for reading the `NetworkService` objects and `MetadataCollection` from our network database.
+
+    NOTE: The network database must be loaded first if you are using a pre-split database you wish to upgrade as it was the only database at the time
+      and will create the other databases as part of the upgrade. This warning can be removed once we set a new minimum version of the database and
+      remove the split database logic - Check `UpgradeRunner` to see if this is still required.
+
+    :param connection: The connection to the database.
+    :param service: The `NetworkService` to populate with CIM objects from the database.
+    :param database_description: The description of the database for logging (e.g. filename).
+    """
+
+    def __init__(
+        self,
+        connection: Connection,
+        service: NetworkService,
+        database_description: str,
+        infer_phases: bool = None,
+        metadata_reader: MetadataCollectionReader = None,
+        service_reader: NetworkServiceReader = None,
+        table_version: TableVersion = TableVersion(),
+        set_feeder_direction: SetDirection = Tracing.set_direction(),
+        set_phases: SetPhases = Tracing.set_phases(),
+        phase_inferrer: PhaseInferrer = Tracing.phase_inferrer(),
+        assign_to_feeders: AssignToFeeders = Tracing.assign_equipment_to_feeders(),
+        assign_to_lv_feeders: AssignToLvFeeders = Tracing.assign_equipment_to_lv_feeders()
+    ):
+        super().__init__(
+            connection,
+            metadata_reader if metadata_reader else MetadataCollectionReader(service, NetworkDatabaseTables(), connection),
+            service_reader if service_reader else NetworkServiceReader(service, NetworkDatabaseTables(), connection),
+            service,
+            database_description,
+            table_version
+        )
+        self.service = service
+        self.infer_phases = infer_phases
+        self.set_feeder_direction = set_feeder_direction
+        self.set_phases = set_phases
+        self.phase_inferrer = phase_inferrer
+        self.assign_to_feeders = assign_to_feeders
+        self.assign_to_lv_feeders = assign_to_lv_feeders
+
+    async def _post_load(self) -> bool:
+        status = await super()._post_load()
+
+        self._logger.info("Applying feeder direction to network...")
+        await self.set_feeder_direction.run(self.service, network_state_operators=NetworkStateOperators.NORMAL)
+        await self.set_feeder_direction.run(self.service, network_state_operators=NetworkStateOperators.CURRENT)
+        self._logger.info("Feeder direction applied to network.")
+
+        self._logger.info("Applying phases to network...")
+        await self.set_phases.run(self.service, network_state_operators=NetworkStateOperators.NORMAL)
+        await self.set_phases.run(self.service, network_state_operators=NetworkStateOperators.CURRENT)
+        if self.infer_phases:
+            self._log_inferred_phases(
+                await self.phase_inferrer.run(self.service, network_state_operators=NetworkStateOperators.NORMAL),
+                await self.phase_inferrer.run(self.service, network_state_operators=NetworkStateOperators.CURRENT)
+            )
+
+        self._logger.info("Phasing applied to network.")
+
+        self._logger.info("Assigning equipment to feeders...")
+        await self.assign_to_feeders.run(self.service, network_state_operators=NetworkStateOperators.NORMAL)
+        await self.assign_to_feeders.run(self.service, network_state_operators=NetworkStateOperators.CURRENT)
+        self._logger.info("Equipment assigned to feeders.")
+
+        self._logger.info("Assigning equipment to LV feeders...")
+        await self.assign_to_lv_feeders.run(self.service, network_state_operators=NetworkStateOperators.NORMAL)
+        await self.assign_to_lv_feeders.run(self.service, network_state_operators=NetworkStateOperators.CURRENT)
+        self._logger.info("Equipment assigned to LV feeders.")
+
+        self._logger.info("Validating that each equipment is assigned to a container...")
+        self._validate_equipment_containers()
+        self._logger.info("Equipment containers validated.")
+
+        self._logger.info("Validating primary sources vs feeders...")
+        self._validate_sources()
+        self._logger.info("Sources vs feeders validated.")
+
+        return status
+
+    def _log_inferred_phases(self,
+                             normal_inferred_phases: List[PhaseInferrer.InferredPhase],
+                             current_inferred_phases: List[PhaseInferrer.InferredPhase]):
+
+        inferred_phases = {item.conducting_equipment: item for item in normal_inferred_phases}
+
+        for it in current_inferred_phases:
+            if it.conducting_equipment in inferred_phases:
+                left = inferred_phases[it.conducting_equipment]
+                inferred_phases[it.conducting_equipment] = left if left.suspect else it
+            else:
+                inferred_phases[it.conducting_equipment] = it
+
+        for phase in inferred_phases.values():
+            self._logger.warning(f"*** Action Required *** {phase.description}")
+
+    def _validate_equipment_containers(self):
+        missing_containers = [it for it in self.service.objects(Equipment) if not it.containers]
+        count_by_class = Counter()
+
+        for it in missing_containers:
+            count_by_class[type(it).__name__] += 1
+
+        for (className, count) in count_by_class:
+            self._logger.warning(f"{count} {className}s were missing an equipment container.")
+
+        if count_by_class:
+            self._logger.warning(f"A total of {len(missing_containers)} equipment had no associated equipment container. Debug logging will show more details.")
+
+        for equipment in missing_containers:
+            self._logger.debug(f"{equipment} was not assigned to any equipment container.")
+
+    def _validate_sources(self):
+        def head_conducting_equipment_mrid(feeder: Feeder) -> Optional[str]:
+            if feeder.normal_head_terminal and feeder.normal_head_terminal.conducting_equipment:
+                return feeder.normal_head_terminal.conducting_equipment.mrid
+            else:
+                return None
+
+        # We do not want to warn about sources attached directly to the feeder start point.
+        feeder_start_points = set(map(head_conducting_equipment_mrid, self.service.objects(Feeder)))
+
+        def has_been_assigned_to_feeder(energy_source: EnergySource) -> bool:
+            return energy_source.is_external_grid \
+                and self._is_on_feeder(energy_source) \
+                and feeder_start_points.isdisjoint({it.to_equip.mrid for it in connected_equipment(energy_source) if it.to_equip})
+
+        for es in self.service.objects(EnergySource):
+            if has_been_assigned_to_feeder(es):
+                self._logger.warning(
+                    f"External grid source {es.name} [{es.mrid}] has been assigned to the following feeders: normal [{[it.mrid for it in es.normal_feeders]}], "
+                    f"current [{[it.mrid for it in es.current_feeders]}]"
+                )
+
+    @staticmethod
+    def _is_on_feeder(energy_source: EnergySource) -> bool:
+        if energy_source.normal_feeders or energy_source.current_feeders:
+            return True
+        else:
+            return False

zepben/ewb/database/sqlite/network/network_database_tables.py

@@ -0,0 +1,242 @@
+#  Copyright 2024 Zeppelin Bend Pty Ltd
+#  This Source Code Form is subject to the terms of the Mozilla Public
+#  License, v. 2.0. If a copy of the MPL was not distributed with this
+#  file, You can obtain one at https://mozilla.org/MPL/2.0/.
+
+__all__ = ["NetworkDatabaseTables"]
+
+from typing import Generator
+
+from zepben.ewb.database.sqlite.common.base_database_tables import BaseDatabaseTables
+from zepben.ewb.database.sqlite.tables.associations.table_asset_organisation_roles_assets import *
+from zepben.ewb.database.sqlite.tables.associations.table_assets_power_system_resources import TableAssetsPowerSystemResources
+from zepben.ewb.database.sqlite.tables.associations.table_battery_units_battery_controls import *
+from zepben.ewb.database.sqlite.tables.associations.table_circuits_substations import *
+from zepben.ewb.database.sqlite.tables.associations.table_circuits_terminals import *
+from zepben.ewb.database.sqlite.tables.associations.table_end_devices_end_device_functions import *
+from zepben.ewb.database.sqlite.tables.associations.table_equipment_equipment_containers import *
+from zepben.ewb.database.sqlite.tables.associations.table_equipment_operational_restrictions import *
+from zepben.ewb.database.sqlite.tables.associations.table_equipment_usage_points import *
+from zepben.ewb.database.sqlite.tables.associations.table_loops_substations import *
+from zepben.ewb.database.sqlite.tables.associations.table_protection_relay_functions_protected_switches import *
+from zepben.ewb.database.sqlite.tables.associations.table_protection_relay_functions_sensors import *
+from zepben.ewb.database.sqlite.tables.associations.table_protection_relay_schemes_protection_relay_functions import *
+from zepben.ewb.database.sqlite.tables.associations.table_synchronous_machines_reactive_capability_curves import \
+    TableSynchronousMachinesReactiveCapabilityCurves
+from zepben.ewb.database.sqlite.tables.associations.table_usage_points_end_devices import *
+from zepben.ewb.database.sqlite.tables.extensions.iec61968.assetinfo.table_reclose_delays import *
+from zepben.ewb.database.sqlite.tables.extensions.iec61968.assetinfo.table_relay_info import *
+from zepben.ewb.database.sqlite.tables.extensions.iec61968.metering.table_pan_demand_response_functions import TablePanDemandResponseFunctions
+from zepben.ewb.database.sqlite.tables.extensions.iec61970.base.core.table_sites import *
+from zepben.ewb.database.sqlite.tables.extensions.iec61970.base.feeder.table_loops import *
+from zepben.ewb.database.sqlite.tables.extensions.iec61970.base.feeder.table_lv_feeders import *
+from zepben.ewb.database.sqlite.tables.extensions.iec61970.base.generation.production.table_ev_charging_units import *
+from zepben.ewb.database.sqlite.tables.extensions.iec61970.base.protection.table_distance_relays import *
+from zepben.ewb.database.sqlite.tables.extensions.iec61970.base.protection.table_protection_relay_function_thresholds import *
+from zepben.ewb.database.sqlite.tables.extensions.iec61970.base.protection.table_protection_relay_function_time_limits import *
+from zepben.ewb.database.sqlite.tables.extensions.iec61970.base.protection.table_protection_relay_schemes import *
+from zepben.ewb.database.sqlite.tables.extensions.iec61970.base.protection.table_protection_relay_systems import *
+from zepben.ewb.database.sqlite.tables.extensions.iec61970.base.protection.table_voltage_relays import *
+from zepben.ewb.database.sqlite.tables.extensions.iec61970.base.wires.table_battery_controls import TableBatteryControls
+from zepben.ewb.database.sqlite.tables.extensions.iec61970.base.wires.table_power_transformer_end_ratings import *
+from zepben.ewb.database.sqlite.tables.iec61968.assetinfo.table_cable_info import *
+from zepben.ewb.database.sqlite.tables.iec61968.assetinfo.table_no_load_tests import *
+from zepben.ewb.database.sqlite.tables.iec61968.assetinfo.table_open_circuit_tests import *
+from zepben.ewb.database.sqlite.tables.iec61968.assetinfo.table_overhead_wire_info import *
+from zepben.ewb.database.sqlite.tables.iec61968.assetinfo.table_power_transformer_info import *
+from zepben.ewb.database.sqlite.tables.iec61968.assetinfo.table_short_circuit_tests import *
+from zepben.ewb.database.sqlite.tables.iec61968.assetinfo.table_shunt_compensator_info import *
+from zepben.ewb.database.sqlite.tables.iec61968.assetinfo.table_switch_info import *
+from zepben.ewb.database.sqlite.tables.iec61968.assetinfo.table_transformer_end_info import *
+from zepben.ewb.database.sqlite.tables.iec61968.assetinfo.table_transformer_tank_info import *
+from zepben.ewb.database.sqlite.tables.iec61968.assets.table_asset_owners import *
+from zepben.ewb.database.sqlite.tables.iec61968.assets.table_streetlights import *
+from zepben.ewb.database.sqlite.tables.iec61968.common.table_location_street_addresses import *
+from zepben.ewb.database.sqlite.tables.iec61968.common.table_locations import *
+from zepben.ewb.database.sqlite.tables.iec61968.common.table_organisations import *
+from zepben.ewb.database.sqlite.tables.iec61968.common.table_position_points import *
+from zepben.ewb.database.sqlite.tables.iec61968.infiec61968.infassetinfo.table_current_transformer_info import *
+from zepben.ewb.database.sqlite.tables.iec61968.infiec61968.infassetinfo.table_potential_transformer_info import *
+from zepben.ewb.database.sqlite.tables.iec61968.infiec61968.infassets.table_poles import *
+from zepben.ewb.database.sqlite.tables.iec61968.metering.table_meters import *
+from zepben.ewb.database.sqlite.tables.iec61968.metering.table_usage_points import *
+from zepben.ewb.database.sqlite.tables.iec61968.operations.table_operational_restrictions import *
+from zepben.ewb.database.sqlite.tables.iec61970.base.auxiliaryequipment.table_current_transformers import *
+from zepben.ewb.database.sqlite.tables.iec61970.base.auxiliaryequipment.table_fault_indicators import *
+from zepben.ewb.database.sqlite.tables.iec61970.base.auxiliaryequipment.table_potential_transformers import *
+from zepben.ewb.database.sqlite.tables.iec61970.base.core.table_base_voltages import *
+from zepben.ewb.database.sqlite.tables.iec61970.base.core.table_connectivity_nodes import *
+from zepben.ewb.database.sqlite.tables.iec61970.base.core.table_curve_data import TableCurveData
+from zepben.ewb.database.sqlite.tables.iec61970.base.core.table_feeders import *
+from zepben.ewb.database.sqlite.tables.iec61970.base.core.table_geographical_regions import *
+from zepben.ewb.database.sqlite.tables.iec61970.base.core.table_sub_geographical_regions import *
+from zepben.ewb.database.sqlite.tables.iec61970.base.core.table_substations import *
+from zepben.ewb.database.sqlite.tables.iec61970.base.core.table_terminals import *
+from zepben.ewb.database.sqlite.tables.iec61970.base.equivalents.table_equivalent_branches import *
+from zepben.ewb.database.sqlite.tables.iec61970.base.generation.production.table_battery_units import *
+from zepben.ewb.database.sqlite.tables.iec61970.base.generation.production.table_photo_voltaic_units import *
+from zepben.ewb.database.sqlite.tables.iec61970.base.generation.production.table_power_electronics_wind_units import *
+from zepben.ewb.database.sqlite.tables.iec61970.base.meas.table_accumulators import *
+from zepben.ewb.database.sqlite.tables.iec61970.base.meas.table_analogs import *
+from zepben.ewb.database.sqlite.tables.iec61970.base.meas.table_controls import *
+from zepben.ewb.database.sqlite.tables.iec61970.base.meas.table_discretes import *
+from zepben.ewb.database.sqlite.tables.iec61970.base.protection.table_current_relays import *
+from zepben.ewb.database.sqlite.tables.iec61970.base.scada.table_remote_controls import *
+from zepben.ewb.database.sqlite.tables.iec61970.base.scada.table_remote_sources import *
+from zepben.ewb.database.sqlite.tables.iec61970.base.wires.table_ac_line_segments import *
+from zepben.ewb.database.sqlite.tables.iec61970.base.wires.table_breakers import *
+from zepben.ewb.database.sqlite.tables.iec61970.base.wires.table_busbar_sections import *
+from zepben.ewb.database.sqlite.tables.iec61970.base.wires.table_clamps import *
+from zepben.ewb.database.sqlite.tables.iec61970.base.wires.table_cuts import *
+from zepben.ewb.database.sqlite.tables.iec61970.base.wires.table_disconnectors import *
+from zepben.ewb.database.sqlite.tables.iec61970.base.wires.table_energy_consumer_phases import *
+from zepben.ewb.database.sqlite.tables.iec61970.base.wires.table_energy_consumers import *
+from zepben.ewb.database.sqlite.tables.iec61970.base.wires.table_energy_source_phases import *
+from zepben.ewb.database.sqlite.tables.iec61970.base.wires.table_energy_sources import *
+from zepben.ewb.database.sqlite.tables.iec61970.base.wires.table_fuses import *
+from zepben.ewb.database.sqlite.tables.iec61970.base.wires.table_ground_disconnectors import *
+from zepben.ewb.database.sqlite.tables.iec61970.base.wires.table_grounding_impedances import TableGroundingImpedances
+from zepben.ewb.database.sqlite.tables.iec61970.base.wires.table_grounds import *
+from zepben.ewb.database.sqlite.tables.iec61970.base.wires.table_jumpers import *
+from zepben.ewb.database.sqlite.tables.iec61970.base.wires.table_junctions import *
+from zepben.ewb.database.sqlite.tables.iec61970.base.wires.table_linear_shunt_compensators import *
+from zepben.ewb.database.sqlite.tables.iec61970.base.wires.table_load_break_switches import *
+from zepben.ewb.database.sqlite.tables.iec61970.base.wires.table_per_length_phase_impedances import TablePerLengthPhaseImpedances
+from zepben.ewb.database.sqlite.tables.iec61970.base.wires.table_per_length_sequence_impedances import *
+from zepben.ewb.database.sqlite.tables.iec61970.base.wires.table_petersen_coils import TablePetersenCoils
+from zepben.ewb.database.sqlite.tables.iec61970.base.wires.table_phase_impedance_data import TablePhaseImpedanceData
+from zepben.ewb.database.sqlite.tables.iec61970.base.wires.table_power_electronics_connection_phases import *
+from zepben.ewb.database.sqlite.tables.iec61970.base.wires.table_power_electronics_connections import *
+from zepben.ewb.database.sqlite.tables.iec61970.base.wires.table_power_transformer_ends import *
+from zepben.ewb.database.sqlite.tables.iec61970.base.wires.table_power_transformers import *
+from zepben.ewb.database.sqlite.tables.iec61970.base.wires.table_ratio_tap_changers import *
+from zepben.ewb.database.sqlite.tables.iec61970.base.wires.table_reactive_capability_curves import TableReactiveCapabilityCurves
+from zepben.ewb.database.sqlite.tables.iec61970.base.wires.table_reclosers import *
+from zepben.ewb.database.sqlite.tables.iec61970.base.wires.table_series_compensators import *
+from zepben.ewb.database.sqlite.tables.iec61970.base.wires.table_static_var_compensator import TableStaticVarCompensators
+from zepben.ewb.database.sqlite.tables.iec61970.base.wires.table_synchronous_machines import TableSynchronousMachines
+from zepben.ewb.database.sqlite.tables.iec61970.base.wires.table_tap_changer_controls import *
+from zepben.ewb.database.sqlite.tables.iec61970.base.wires.table_transformer_star_impedances import *
+from zepben.ewb.database.sqlite.tables.iec61970.infiec61970.feeder.table_circuits import *
+from zepben.ewb.database.sqlite.tables.sqlite_table import *
+
+
+class NetworkDatabaseTables(BaseDatabaseTables):
+    """
+    The collection of tables for our customer databases.
+    """
+
+    @property
+    def _included_tables(self) -> Generator[SqliteTable, None, None]:
+        for table in super()._included_tables:
+            yield table
+
+        yield TableAcLineSegments()
+        yield TableAccumulators()
+        yield TableAnalogs()
+        yield TableAssetOrganisationRolesAssets()
+        yield TableAssetsPowerSystemResources()
+        yield TableAssetOwners()
+        yield TableBaseVoltages()
+        yield TableBatteryControls()
+        yield TableBatteryUnits()
+        yield TableBatteryUnitsBatteryControls()
+        yield TableBreakers()
+        yield TableBusbarSections()
+        yield TableCableInfo()
+        yield TableCircuits()
+        yield TableCircuitsSubstations()
+        yield TableCircuitsTerminals()
+        yield TableClamps()
+        yield TableConnectivityNodes()
+        yield TableControls()
+        yield TableCurrentRelays()
+        yield TableCurrentTransformerInfo()
+        yield TableCurrentTransformers()
+        yield TableCurveData()
+        yield TableCuts()
+        yield TableDisconnectors()
+        yield TableDiscretes()
+        yield TableDistanceRelays()
+        yield TableEndDevicesEndDeviceFunctions()
+        yield TableEnergyConsumerPhases()
+        yield TableEnergyConsumers()
+        yield TableEnergySourcePhases()
+        yield TableEnergySources()
+        yield TableEquipmentEquipmentContainers()
+        yield TableEquipmentOperationalRestrictions()
+        yield TableEquipmentUsagePoints()
+        yield TableEquivalentBranches()
+        yield TableEvChargingUnits()
+        yield TableFaultIndicators()
+        yield TableFeeders()
+        yield TableFuses()
+        yield TableGeographicalRegions()
+        yield TableGrounds()
+        yield TableGroundDisconnectors()
+        yield TableGroundingImpedances()
+        yield TableJumpers()
+        yield TableJunctions()
+        yield TableLinearShuntCompensators()
+        yield TableLoadBreakSwitches()
+        yield TableLocationStreetAddresses()
+        yield TableLocations()
+        yield TableLoops()
+        yield TableLoopsSubstations()
+        yield TableLvFeeders()
+        yield TableMeters()
+        yield TableNoLoadTests()
+        yield TableOpenCircuitTests()
+        yield TableOperationalRestrictions()
+        yield TableOrganisations()
+        yield TableOverheadWireInfo()
+        yield TablePanDemandResponseFunctions()
+        yield TablePerLengthPhaseImpedances()
+        yield TablePerLengthSequenceImpedances()
+        yield TablePhaseImpedanceData()
+        yield TablePetersenCoils()
+        yield TablePhotoVoltaicUnits()
+        yield TablePoles()
+        yield TablePositionPoints()
+        yield TablePotentialTransformerInfo()
+        yield TablePotentialTransformers()
+        yield TablePowerElectronicsConnections()
+        yield TablePowerElectronicsConnectionPhases()
+        yield TablePowerElectronicsWindUnits()
+        yield TablePowerTransformerEnds()
+        yield TablePowerTransformerEndRatings()
+        yield TablePowerTransformerInfo()
+        yield TablePowerTransformers()
+        yield TableProtectionRelayFunctionThresholds()
+        yield TableProtectionRelayFunctionTimeLimits()
+        yield TableProtectionRelayFunctionsProtectedSwitches()
+        yield TableProtectionRelayFunctionsSensors()
+        yield TableProtectionRelaySchemes()
+        yield TableProtectionRelaySchemesProtectionRelayFunctions()
+        yield TableProtectionRelaySystems()
+        yield TableRatioTapChangers()
+        yield TableReactiveCapabilityCurves()
+        yield TableReclosers()
+        yield TableRecloseDelays()
+        yield TableRelayInfo()
+        yield TableRemoteControls()
+        yield TableRemoteSources()
+        yield TableSeriesCompensators()
+        yield TableShortCircuitTests()
+        yield TableShuntCompensatorInfo()
+        yield TableSites()
+        yield TableStaticVarCompensators()
+        yield TableStreetlights()
+        yield TableSubGeographicalRegions()
+        yield TableSubstations()
+        yield TableSwitchInfo()
+        yield TableSynchronousMachines()
+        yield TableSynchronousMachinesReactiveCapabilityCurves()
+        yield TableTapChangerControls()
+        yield TableTerminals()
+        yield TableTransformerEndInfo()
+        yield TableTransformerStarImpedances()
+        yield TableTransformerTankInfo()
+        yield TableUsagePoints()
+        yield TableUsagePointsEndDevices()
+        yield TableVoltageRelays()

zepben/ewb/database/sqlite/network/network_database_writer.py

@@ -0,0 +1,43 @@
+#  Copyright 2024 Zeppelin Bend Pty Ltd
+#  This Source Code Form is subject to the terms of the Mozilla Public
+#  License, v. 2.0. If a copy of the MPL was not distributed with this
+#  file, You can obtain one at https://mozilla.org/MPL/2.0/.
+
+__all__ = ["NetworkDatabaseWriter"]
+
+import sqlite3
+from pathlib import Path
+from sqlite3 import Connection
+from typing import Callable, Union
+
+from zepben.ewb.database.sqlite.common.base_database_writer import BaseDatabaseWriter
+from zepben.ewb.database.sqlite.common.metadata_collection_writer import MetadataCollectionWriter
+from zepben.ewb.database.sqlite.network.network_database_tables import NetworkDatabaseTables
+from zepben.ewb.database.sqlite.network.network_service_writer import NetworkServiceWriter
+from zepben.ewb.services.network.network_service import NetworkService
+
+
+class NetworkDatabaseWriter(BaseDatabaseWriter):
+    """
+    A class for writing the `NetworkService` objects and `MetadataCollection` to our network database.
+
+    :param database_file: the filename of the database to write.
+    :param service: The `NetworkService` to save to the database.
+    """
+
+    def __init__(
+        self,
+        database_file: Union[Path, str],
+        service: NetworkService,
+        database_tables: NetworkDatabaseTables = NetworkDatabaseTables(),
+        create_metadata_writer: Callable[[Connection], MetadataCollectionWriter] = None,
+        create_service_writer: Callable[[Connection], NetworkServiceWriter] = None,
+        get_connection: Callable[[str], Connection] = None
+    ):
+        super().__init__(
+            database_file,
+            database_tables,
+            create_metadata_writer if create_metadata_writer is not None else lambda: MetadataCollectionWriter(service, database_tables),
+            create_service_writer if create_service_writer is not None else lambda: NetworkServiceWriter(service, database_tables),
+            get_connection if get_connection is not None else sqlite3.connect
+        )