Salome HOME
Add white list for redirect some missing attributes to the feature.
[modules/shaper.git] / src / PythonAPI / model / tools.py
1 """Common tools for other modules.
2 Author: Sergey Pokhodenko
3 Copyright (C) 2014-20xx CEA/DEN, EDF R&D
4 """
5
6 import re
7 import ModelAPI
8
9 def convert_to_underscore(name):
10     """Convert CamelCase to underscore_case."""
11     s1 = re.sub('(.)([A-Z][a-z]+)', r'\1_\2', name)
12     return re.sub('([a-z0-9])([A-Z])', r'\1_\2', s1).lower()
13
14 def get_value(attribute):
15     """Return simple type value from ModelAPI attribute if possible.
16     
17     ModelAPI_AttributeDouble -> double
18     ModelAPI_AttributeString -> str
19     """
20     if (isinstance(attribute, ModelAPI.ModelAPI_AttributeDouble) or
21             isinstance(attribute, ModelAPI.ModelAPI_AttributeString)):
22         return attribute.value()
23
24     print type(attribute)
25     return attribute