def create_new_element_group(self, title = None):
self.title = title
if title is None:
self.element_group_model = ElementGroupModel()
else:
self.element_group_model = ElementGroupModel(title = title)
self.element_group_model.save()
class ElementGroup(object):
def __init__(self, title = None, element_group_model = None):
self.elements = []
if (element_group_model is None):
self.create_new_element_group(title = title)
else:
self.load_element_group_from_database(element_group_model)
def create_new_element_group(self, title = None):
self.title = title
if title is None:
self.element_group_model = ElementGroupModel()
else:
self.element_group_model = ElementGroupModel(title = title)
self.element_group_model.save()
def load_element_group_from_database(self, element_group_model = None):
self.element_group_model = element_group_model
self.title = self.element_group_model.title
def add_element(self, element):
element.set_element_group(self)
self.elements.append(element)
def get_elements_of_type(self, type):
result = []
for element in self.elements:
if isinstance(element, type):
result.append(element)
return result
