feat:kube session attributes

Author: Turtdle

src/pybritive/britive_cli.py

@@ -486,6 +486,18 @@ def _get_missing_env_properties(
             )
         return {}
 
+    def _fetch_session_attributes_for_profiles(self, accesses):
+        session_attributes = {}
+        unique_profile_ids = {profile_id for _, _, profile_id in accesses}
+        for profile_id in unique_profile_ids:
+            try:
+                session_attrs = self.b.application_management.profiles.session_attributes.list(profile_id=profile_id)
+                session_attributes[profile_id] = session_attrs
+            except Exception:
+                session_attributes[profile_id] = {}
+
+        return session_attributes
+
     def _set_available_profiles(self, from_cache_command=False, profile_type: Optional[str] = None):
         if not self.available_profiles:
             data = []
@@ -503,6 +515,7 @@ def _set_available_profiles(self, from_cache_command=False, profile_type: Option
                 accesses = [
                     ([a['appContainerId'], a['environmentId'], a['papId']]) for a in access_data.get('accesses', [])
                 ]
+                session_attributes = self._fetch_session_attributes_for_profiles(accesses)
                 access_output = []
                 for app_id, env_id, profile_id in accesses:
                     app = apps[app_id]
@@ -522,6 +535,7 @@ def _set_available_profiles(self, from_cache_command=False, profile_type: Option
                         'profile_allows_console': app.get('consoleAccess', False),
                         'profile_allows_programmatic': app.get('programmaticAccess', False),
                         'profile_description': profile['papDescription'],
+                        'session_attributes': session_attributes.get(profile_id, []),
                         '2_part_profile_format_allowed': app['requiresHierarchicalModel'],
                         'env_properties': env['profileEnvironmentProperties']
                         or self._get_missing_env_properties(
@@ -537,6 +551,7 @@ def _set_available_profiles(self, from_cache_command=False, profile_type: Option
                 else:
                     profiles = self.b.my_resources.list(size=resource_limit)
                     profiles = profiles['data']
+                session_attributes = self._fetch_session_attributes_for_profiles(accesses)
                 for item in profiles:
                     row = {
                         'app_name': None,
@@ -551,6 +566,7 @@ def _set_available_profiles(self, from_cache_command=False, profile_type: Option
                         'profile_id': item['profileId'],
                         'profile_allows_console': False,
                         'profile_allows_programmatic': True,
+                        'session_attributes': session_attributes.get(profile_id, []),
                         'profile_description': None,
                         '2_part_profile_format_allowed': False,
                         'env_properties': item.get('resourceLabels', {}),
@@ -584,6 +600,7 @@ def construct_kube_config(self, from_cache_command=False):
                             'profile': p['profile_name'],
                             'url': url,
                             'cert': cert,
+                            'session_attributes': p['session_attributes'],
                         }
                     )
 

src/pybritive/helpers/kube_config_builder.py

@@ -69,7 +69,7 @@ def parse_profiles(profiles, aliases):
     cluster_names = {}
     assigned_aliases = []
     for profile in profiles:
-        env_profile = f"{sanitize(profile['env'])}-{sanitize(profile['profile'].lower())}"
+        env_profile = f'{sanitize(profile["env"])}-{sanitize(profile["profile"].lower())}'
         if env_profile not in cluster_names:
             app = BritiveCli.escape_profile_element(profile['app'])
             env = BritiveCli.escape_profile_element(profile['env'])
@@ -78,14 +78,17 @@ def parse_profiles(profiles, aliases):
             escaped_profile_str = f'{app}/{env}/{pro}'.lower()
             alias = aliases.get(escaped_profile_str, None)
             assigned_aliases.append(alias)
+            keys_to_remove = ['id', 'attributeSchemaId', 'transitive', 'sessionAttributeType']
+            [list(map(attr.pop, keys_to_remove)) for attr in profile['session_attributes']]
 
             cluster_names[env_profile] = {
                 'apps': [],
                 'url': profile['url'],
                 'cert': profile['cert'],
                 'escaped_profile': escaped_profile_str,
-                'profile': f"{profile['app']}/{profile['env']}/{profile['profile']}".lower(),
+                'profile': f'{profile["app"]}/{profile["env"]}/{profile["profile"]}'.lower(),
                 'alias': alias,
+                'session_attributes': profile['session_attributes'],
             }
         cluster_names[env_profile]['apps'].append(sanitize(profile['app']))
     return [cluster_names, assigned_aliases]
@@ -129,7 +132,6 @@ def build_tenant_config(tenant, cluster_names, username, cli: BritiveCli):
     )
     contexts = []
     clusters = []
-
     for env_profile, details in cluster_names.items():
         if len(details['apps']) == 1:
             names = [env_profile]
@@ -138,6 +140,7 @@ def build_tenant_config(tenant, cluster_names, username, cli: BritiveCli):
 
         cert = details['cert']
         url = details['url']
+        session_attributes = details['session_attributes']
 
         if not valid_cert(cert=cert, profile=details['profile'], cli=cli):
             continue
@@ -162,7 +165,11 @@ def build_tenant_config(tenant, cluster_names, username, cli: BritiveCli):
             contexts.append(
                 {
                     'name': details.get('alias') or f'{tenant}-{name}',
-                    'context': {'cluster': f'{tenant}-{name}', 'user': username},
+                    'context': {
+                        'cluster': f'{tenant}-{name}',
+                        'user': username,
+                        **{attr['mappingName']: attr['attributeValue'] for attr in session_attributes},
+                    },
                 }
             )
     return [clusters, contexts, users]