Added custom expansion support for repeated capabilities

Author: Vaishnavigupta1312

build/templates/session.py.mako

@@ -88,7 +88,6 @@ class _Lock(object):
     def __exit__(self, exc_type, exc_value, traceback):
         self._session.unlock()
 
-
 % endif
 % if len(config['repeated_capabilities']) > 0:
 
@@ -205,6 +204,7 @@ class _NoChannel(object):
 
     def __exit__(self, exc_type, exc_value, traceback):
         self._session._repeated_capability = self._repeated_capability_cache
+% endif
 
 
 % endif
@@ -216,6 +216,11 @@ class _SessionBase(object):
 
 % for attribute in helper.sorted_attrs(helper.filter_codegen_attributes(attributes)):
 <%
+# Skip attributes with repeated capability expansion set to "custom"
+if 'repeated_capability_type' in attributes[attribute]:
+    rep_cap_type = attributes[attribute]['repeated_capability_type']
+    if any(rep_cap.get('python_name') == rep_cap_type and config['rep_cap_expansion'] == 'custom' for rep_cap in config['repeated_capabilities']):
+        continue
 helper.add_attribute_rep_cap_tip(attributes[attribute], config)
 %>\
     %if attributes[attribute]['enum']:
@@ -257,9 +262,13 @@ constructor_params = helper.filter_parameters(init_function['parameters'], helpe
 
 % if len(config['repeated_capabilities']) > 0:
         # Instantiate any repeated capability objects
-%   for rep_cap in config['repeated_capabilities']:
+% for rep_cap in config['repeated_capabilities']:
+% if config['rep_cap_expansion'] == 'custom':
+        self.${rep_cap['python_name']} = _RepCap${rep_cap['python_name'].capitalize()}(self, repeated_capability_list)
+% else:
         self.${rep_cap['python_name']} = _RepeatedCapabilities(self, '${rep_cap["prefix"]}', repeated_capability_list)
-%   endfor
+% endif
+% endfor
 
 % endif
         # Finally, set _is_frozen to True which is used to prevent clients from accidentally adding

generated/nidmm/nidmm/session.py

@@ -50,7 +50,6 @@ def __enter__(self):
     def __exit__(self, exc_type, exc_value, traceback):
         self._session.unlock()
 
-
 class _SessionBase(object):
     '''Base class for all NI-DMM sessions.'''