File size: 509 Bytes
2795186
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
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 }