def translate_auth_to_avro(*, auth: AuthToken) -> AuthAvro: result = AuthAvro() result.name = auth.get_name() result.guid = str(auth.get_guid()) result.oidc_sub_claim = auth.get_oidc_sub_claim() result.email = auth.get_email() return result
def set_identity(self, *, token: AuthToken): """ Set identity @param token token """ self.identity = token self.name = self.identity.get_name() self.guid = token.get_guid()
def __init__(self, *, auth: AuthToken = None): self.logger = None self.proxy_type = None self.callback = False self.actor_name = None if auth is not None: self.actor_name = auth.get_name() self.actor_guid = None if auth is not None: self.actor_guid = auth.get_guid() self.auth = auth
class ActorIdentity(ABCActorIdentity): """ Represent Actor Identity """ def __init__(self, *, name: str, guid: ID): self.auth = AuthToken(name=name, guid=guid) def get_guid(self) -> ID: """ Return guid @return guid """ return self.auth.get_guid() def get_identity(self) -> AuthToken: """ Return identity @return identity """ return self.auth def get_name(self) -> str: return self.auth.get_name()
class ClientCallbackHelper(ABCClientCallbackProxy): class MyRequestState(RPCRequestState): def __init__(self): super().__init__() self.reservation = None self.update_data = None def __init__(self, *, name: str, guid: ID): self.token = AuthToken(name=name, guid=guid) self.called = 0 self.prepared = 0 self.reservation = None def get_called(self) -> int: return self.called def get_guid(self) -> ID: return self.token.get_guid() def get_identity(self) -> AuthToken: return self.token def get_name(self) -> str: return self.token.get_name() def get_reservation(self) -> ABCReservationMixin: return self.reservation def get_type(self): return Constants.PROTOCOL_LOCAL def prepare_update_delegation(self, *, delegation: ABCDelegation, update_data: UpdateData, callback: ABCCallbackProxy, caller: AuthToken) -> ABCRPCRequestState: raise NotImplementedError def set_logger(self, *, logger): pass def prepare_update_ticket(self, *, reservation: ABCBrokerReservation, update_data: UpdateData, callback: ABCCallbackProxy, caller: AuthToken) -> ABCRPCRequestState: state = self.MyRequestState() state.reservation = LocalReturn.pass_reservation(reservation=reservation, plugin=ActorRegistrySingleton.get().get_actor( actor_name_or_guid=self.token.get_name()).get_plugin()) self.prepared += 1 return state def execute(self, *, request: ABCRPCRequestState, producer: AvroProducerApi): if request.get_type() == RPCRequestType.UpdateTicket: self.called += 1 self.reservation = request.reservation def get_logger(self): from fabric_cf.actor.core.container.globals import GlobalsSingleton return GlobalsSingleton.get().get_logger() def prepare_query_result(self, *, request_id: str, response, caller: AuthToken) -> ABCRPCRequestState: raise NotImplementedError def prepare_failed_request(self, *, request_id: str, failed_request_type, failed_reservation_id: ID, error: str, caller: AuthToken) -> ABCRPCRequestState: raise NotImplementedError