"""Producer Helpers."""

from time import time

from confluent_kafka.serialization import MessageField


def get_context(context):
    """Set serialization context."""
    message_field = MessageField()
    if context == 'value':
        msg_ctx = message_field.VALUE
    elif context == 'key':
        msg_ctx = message_field.KEY
    else:
        raise AttributeError('Context must be "value" or "key"')
    return msg_ctx


def build_product_event(operation_type, operation_context, event_payload):
    """Build a product event message."""
    return {
        'operation': {
            'type': operation_type,
            'timestamp': time() * 1000,
            'context': operation_context
        },
        'payload': event_payload
    }