Skip to content
Snippets Groups Projects
forward_evidence.py 787 B
Newer Older
Matevz Erzen's avatar
Matevz Erzen committed
from evidence.evidence_store_pb2_grpc import EvidenceStoreStub
from evidence.evidence_pb2 import Evidence
import grpc

class ForwardEvidence(object):

Zitnik, Anze's avatar
Zitnik, Anze committed
    def __init__(self, constants, logger):
Matevz Erzen's avatar
Matevz Erzen committed
        self.channel = grpc.insecure_channel('{}:{}'.format(constants['clouditor']['host'], constants['clouditor']['port']))
        self.stub = EvidenceStoreStub(self.channel)
Zitnik, Anze's avatar
Zitnik, Anze committed
        self.logger = logger
Matevz Erzen's avatar
Matevz Erzen committed

    def send_evidence(self, evidence):
        try:
            response = self.stub.StoreEvidence(evidence)  
Zitnik, Anze's avatar
Zitnik, Anze committed
            self.logger.info('gRPC evidence forwarded: ' + str(response))
Matevz Erzen's avatar
Matevz Erzen committed
        except grpc.RpcError as err:
Zitnik, Anze's avatar
Zitnik, Anze committed
            self.logger.error(err)
            self.logger.error(err.details())
            self.logger.error('{}, {}'.format(err.code().name, err.code().value))
Matevz Erzen's avatar
Matevz Erzen committed

Matevz Erzen's avatar
Matevz Erzen committed