Materialize/misc/python/materialize/zippy/kafka_capabilities.py

# Copyright Materialize, Inc. and contributors. All rights reserved.
#
# Use of this software is governed by the Business Source License
# included in the LICENSE file at the root of this repository.
#
# As of the Change Date specified in that file, in accordance with
# the Business Source License, use of this software will be governed
# by the Apache License, Version 2.0.

from enum import Enum

from materialize.zippy.framework import Capability
from materialize.zippy.watermarks import Watermarks


class KafkaRunning(Capability):
    """Kafka is running in the environment."""

    pass


class Envelope(Enum):
    """Kafka envelope to be used for a particular topic or source.

    If the Envelope is NONE, no deletions take place on the topic, just insertions
    """

    NONE = 1
    UPSERT = 2


class TopicExists(Capability):
    """A Topic exists on the Kafka instance."""

    @classmethod
    def format_str(cls) -> str:
        return "topic-{}"

    def __init__(self, name: str, partitions: int, envelope: Envelope) -> None:
        self.name = name
        self.partitions = partitions
        self.envelope = envelope
        self.watermarks = Watermarks()

    def get_watermarks(self) -> Watermarks:
        return self.watermarks