from dmmc.grpc.media_app_pb2 import MediaBulkRetrieveRequest, MediaRetrieveRequest
from dmmc.grpc.media_app_pb2_grpc import MediaControllerStub

from .config import CHANNEL

SUCCESS_CODE = "0"


class RequestMessages:
    FETCH_ONE_MEDIA = MediaRetrieveRequest
    FETCH_BULK_MEDIA = MediaBulkRetrieveRequest


class RPCMethodNames:
    FETCH_ONE = "Retrieve"
    FETCH_BULK = "BulkRetrieve"


class Stubs:
    MEDIA = MediaControllerStub(channel=CHANNEL)
