hub/catalog_factories/data_models/energy_systems/content.py

@@ -69,11 +69,4 @@ class Content:
 
     return content
 
-  def __str__(self):
-    """Print content"""
-    _archetypes = []
-    for _archetype in self.archetypes:
-      _archetypes.append(_archetype.to_dictionary())
-    content = {'Archetypes': _archetypes}
 
-    return str(content)

hub/catalog_factories/data_models/energy_systems/system.py

@@ -9,8 +9,7 @@ Code contributors: Saeed Ranjbar saeed.ranjbar@concordia.ca
 from typing import Union, List
 
 from hub.catalog_factories.data_models.energy_systems.generation_system import GenerationSystem
-from hub.catalog_factories.data_models.energy_systems.thermal_storage_system import ThermalStorageSystem
-from hub.catalog_factories.data_models.energy_systems.electrical_storage_system import ElectricalStorageSystem
+from hub.catalog_factories.data_models.energy_systems.energy_storage_system import EnergyStorageSystem
 from hub.catalog_factories.data_models.energy_systems.distribution_system import DistributionSystem
 from hub.catalog_factories.data_models.energy_systems.emission_system import EmissionSystem
 
@@ -96,7 +95,7 @@ class System:
     return self._emission_systems
 
   @property
-  def energy_storage_systems(self) -> Union[None, List[ThermalStorageSystem], List[ElectricalStorageSystem]]:
+  def energy_storage_systems(self) -> Union[None, List[EnergyStorageSystem]]:
     """
     Get energy storage systems
     :return: [EnergyStorageSystem]

hub/catalog_factories/data_models/energy_systems/thermal_storage_system.py

@@ -67,8 +67,6 @@ class ThermalStorageSystem(EnergyStorageSystem):
 
   def to_dictionary(self):
     """Class content to dictionary"""
-    _layers = None
-    if self.layers is not None:
     _layers = []
     for _layer in self.layers:
       _layers.append(_layer.to_dictionary())

hub/catalog_factories/energy_systems/montreal_custom_catalog.py

@@ -162,33 +162,33 @@ class MontrealCustomCatalog(Catalog):
       name = system['name']
       demands = system['demands']['demand']
       generation_equipment = system['equipments']['generation_id']
-      _generation_equipments = None
+      _generation_equipment = None
       for equipment_archetype in self._catalog_generation_equipments:
         if int(equipment_archetype.id) == int(generation_equipment):
-          _generation_equipments = [equipment_archetype]
-      _storage_equipments = None
+          _generation_equipment = equipment_archetype
+      _storage_equipment = None
       for equipment_archetype in self._catalog_storage_equipments:
         if int(equipment_archetype.id) == int(generation_equipment):
-          _storage_equipments = [equipment_archetype]
+          _storage_equipment = equipment_archetype
       distribution_equipment = system['equipments']['distribution_id']
-      _distribution_equipments = None
+      _distribution_equipment = None
       for equipment_archetype in self._catalog_distribution_equipments:
         if int(equipment_archetype.id) == int(distribution_equipment):
-          _distribution_equipments = [equipment_archetype]
+          _distribution_equipment = equipment_archetype
       emission_equipment = system['equipments']['dissipation_id']
-      _emission_equipments = None
+      _emission_equipment = None
       for equipment_archetype in self._catalog_emission_equipments:
         if int(equipment_archetype.id) == int(emission_equipment):
-          _emission_equipments = [equipment_archetype]
+          _emission_equipment = equipment_archetype
 
       _catalog_systems.append(System(self._lod,
                                      system_id,
                                      name,
                                      demands,
-                                     _generation_equipments,
-                                     _distribution_equipments,
-                                     _emission_equipments,
-                                     _storage_equipments))
+                                     [_generation_equipment],
+                                     [_distribution_equipment],
+                                     [_emission_equipment],
+                                     [_storage_equipment]))
     return _catalog_systems
 
   def _load_archetypes(self):

hub/catalog_factories/energy_systems/north_america_energy_system_catalog.py

@@ -38,7 +38,7 @@ class NorthAmericaEnergySystemCatalog(Catalog):
                             None,
                             None,
                             self._storage_components)
-
+    print(self._content)
   def _load_generation_components(self):
     generation_components = []
     boilers = self._archetypes['EnergySystemCatalog']['energy_generation_components']['boilers']