PySimultan 0.6.0.3__py3-none-any.whl → 0.6.0.7__py3-none-any.whl

PySimultan2/CHANGELOG.md

@@ -1,3 +1,6 @@
+Version 0.6.0.3 (23.01.2024)
+- Added sub_classes and super_classes to SimultanObject
+
 Version 0.6.0.2 (16.01.2024)
 - Added __delitem__ method to default_types.ComponentList
 

PySimultan2/__about__.py

@@ -1 +1 @@
-version = '0.6.0.3'
+version = '0.6.0.7'

PySimultan2/default_types.py

@@ -5,6 +5,7 @@ import numpy as np
 import colorlog
 from typing import Union, List, Type, Set, Tuple, Any, Optional
 
+from .geometry.geometry_base import classproperty
 from .utils import (sort_slots, create_simultan_component_for_taxonomy, create_mapped_python_object,
                     set_property_to_sim_component, set_property_to_parameter, set_property_to_file_info,
                     set_property_to_list, set_property_to_value_field, set_property_to_unknown_type,
@@ -32,6 +33,10 @@ class ComponentList(SimultanObject):
     _create_all = False     # if true all properties are evaluated to create python objects when initialized
     _taxonomy = 'ComponentList'
 
+    @classproperty
+    def _original_class(cls) -> Type[ComponentList]:
+        return cls
+
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)
         self.component_policy = kwargs.get('component_policy', 'subcomponent')  # component add policy of the content/parameter/property, 'reference' or 'subcomponent'
@@ -405,6 +410,10 @@ class ComponentDictionary(SimultanObject):
 
     _taxonomy = 'ComponentDict'
 
+    @classproperty
+    def _original_class(cls) -> Type[ComponentList]:
+        return cls
+
     def __init__(self, *args, **kwargs):
         self._dict = {}
         super().__init__(*args, **kwargs)

PySimultan2/object_mapper.py

@@ -168,6 +168,7 @@ class PythonMapper(object):
                           '_cls_instances': WeakSet(),
                           '_taxonomy_map': self.taxonomy_maps.get(taxonomy, None),
                           '_base': bases,
+                          '_original_cls': cls,
                           '_object_mapper': self}
 
         new_class_dict.update(self.get_properties(taxonomy))

PySimultan2/simultan_object.py

@@ -130,6 +130,42 @@ class SimultanObject(object, metaclass=MetaMock):
             logger.error(f'Error getting cls_instances: {e}')
             return []
 
+    @classproperty
+    def super_classes(cls):
+        superclasses = []
+        for key, mcls in cls._object_mapper.mapped_classes.items():
+            if mcls is cls:
+                continue
+            if set(cls._original_cls.__mro__) & set(mcls._original_cls.__mro__) - set([object]) and \
+                    set(mcls._original_cls.__mro__) - set(cls._original_cls.__mro__) - set((SimultanObject,
+                                                           object)):
+                superclasses.append(mcls)
+        return superclasses
+
+    @classproperty
+    def sub_classes(cls):
+        subclasses = []
+        for key, mcls in cls._object_mapper.mapped_classes.items():
+            if mcls is cls:
+                continue
+            if set(cls._original_cls.__mro__) & set(mcls._original_cls.__mro__) - set([object]) and \
+                    set(cls._original_cls.__mro__) - set(mcls._original_cls.__mro__) - set((SimultanObject,
+                                                                                            object)):
+                subclasses.append(mcls)
+        return subclasses
+
+    @classproperty
+    def super_class_instances(cls) -> set['SimultanObject']:
+        instances = set()
+        _ = [instances.update(x.cls_instances) for x in cls.super_classes]
+        return instances
+
+    @classproperty
+    def sub_class_instances(cls) -> set['SimultanObject']:
+        instances = set()
+        _ = [instances.update(x.cls_instances) for x in cls.sub_classes]
+        return instances
+
     @classmethod
     def create_simultan_component(cls, *args, **kwargs) -> SimComponent:
         wrapped_obj = utils.create_simultan_component_for_taxonomy(cls, *args, **kwargs)