core: models: add nested endpoints to models

1 files changed, 4 insertions, 0 deletions

diff --git a/squad_client/core/models.py b/squad_client/core/models.py
index 3ffd037..6bf0479 100644
--- a/squad_client/core/models.py
+++ b/squad_client/core/models.py
@@ -51,6 +51,10 @@ class SquadObject:
         if obj is None:
             obj = self
 
+        nested_endpoints = [x[1] for x in SquadApi.schema_nested_eps if x[0].lower() == obj.__class__.__name__.lower()]
+        if nested_endpoints:
+            setattr(obj, 'nested_endpoints', nested_endpoints)
+
         attrs = obj.attrs if len(obj.attrs) else [attr for attr in result.keys()]
         for attr in attrs:
             setattr(obj, attr.replace(' ', '_').replace('/', '_').replace('-', '_'), result[attr])