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