29 lines
916 B
Python
29 lines
916 B
Python
from colony.core.network.nodes import LocalNode
|
|
from colony.role_registry import RoleRegistry
|
|
|
|
|
|
class RoleContext:
|
|
def __init__(self, network, role_manager):
|
|
self._network = network
|
|
self._role_manager = role_manager
|
|
|
|
def execute(self, command, *args, **kwargs):
|
|
return self._network.execute(command, *args, **kwargs)
|
|
|
|
def execute_to_unregistered_node(self, ip, port, command, *args, **kwargs):
|
|
return self._network.execute_to_unregistered_node(ip, port, command, *args, **kwargs)
|
|
|
|
def peers(self):
|
|
return self._network.get_all_nodes()
|
|
|
|
def self(self) -> LocalNode:
|
|
return self._network.get_self()
|
|
|
|
def set_self_node_roles(self, roles):
|
|
self._role_manager.set_roles(roles)
|
|
|
|
def get_role_manager(self):
|
|
return self._role_manager
|
|
|
|
def get_role_registry(self) -> RoleRegistry:
|
|
return self._role_manager.role_registry |