scripts/amlsim/normal_model.py

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 }