| class NormalModel: | |
| def __init__(self, id, type, node_ids, main_id): | |
| self.id = id | |
| self.type = type | |
| self.node_ids = node_ids or set() | |
| self.main_id = main_id | |
| def add_account(self, id): | |
| self.node_ids.add(id) | |
| def is_main(self, node_id): | |
| return node_id == self.main_id | |
| def remove_node_ids(self, node_ids): | |
| self.node_ids = self.node_ids - node_ids | |
| def node_ids_without_main(self): | |
| return self.node_ids - { self.main_id } | |