"""
@generated by mypy-protobuf.  Do not edit manually!
isort:skip_file
"""

import builtins
import google.protobuf.descriptor
import google.protobuf.internal.enum_type_wrapper
import google.protobuf.message
import senren.feature_store.v1.specs.metadata_pb2
import sys
import typing

if sys.version_info >= (3, 10):
    import typing as typing_extensions
else:
    import typing_extensions

DESCRIPTOR: google.protobuf.descriptor.FileDescriptor

class _EntityType:
    ValueType = typing.NewType("ValueType", builtins.int)
    V: typing_extensions.TypeAlias = ValueType

class _EntityTypeEnumTypeWrapper(
    google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_EntityType.ValueType], builtins.type
):
    DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
    STRING: _EntityType.ValueType  # 0
    INT32: _EntityType.ValueType  # 1
    INT64: _EntityType.ValueType  # 2

class EntityType(_EntityType, metaclass=_EntityTypeEnumTypeWrapper): ...

STRING: EntityType.ValueType  # 0
INT32: EntityType.ValueType  # 1
INT64: EntityType.ValueType  # 2
global___EntityType = EntityType

@typing.final
class EntitySpec(google.protobuf.message.Message):
    DESCRIPTOR: google.protobuf.descriptor.Descriptor

    METADATA_FIELD_NUMBER: builtins.int
    JOIN_KEY_FIELD_NUMBER: builtins.int
    TYPE_FIELD_NUMBER: builtins.int
    join_key: builtins.str
    type: global___EntityType.ValueType
    @property
    def metadata(self) -> senren.feature_store.v1.specs.metadata_pb2.CommonMetadata: ...
    def __init__(
        self,
        *,
        metadata: senren.feature_store.v1.specs.metadata_pb2.CommonMetadata | None = ...,
        join_key: builtins.str = ...,
        type: global___EntityType.ValueType = ...,
    ) -> None: ...
    def HasField(self, field_name: typing.Literal["metadata", b"metadata"]) -> builtins.bool: ...
    def ClearField(
        self, field_name: typing.Literal["join_key", b"join_key", "metadata", b"metadata", "type", b"type"]
    ) -> None: ...

global___EntitySpec = EntitySpec

@typing.final
class EntityRef(google.protobuf.message.Message):
    DESCRIPTOR: google.protobuf.descriptor.Descriptor

    NAME_FIELD_NUMBER: builtins.int
    name: builtins.str
    def __init__(
        self,
        *,
        name: builtins.str = ...,
    ) -> None: ...
    def ClearField(self, field_name: typing.Literal["name", b"name"]) -> None: ...

global___EntityRef = EntityRef
