otdf-python 0.1.10__py3-none-any.whl → 0.3.5__py3-none-any.whl

otdf_python_proto/authorization/v2/authorization_pb2_connect.py

@@ -0,0 +1,233 @@
+# Generated Connect client code
+
+from __future__ import annotations
+from collections.abc import AsyncIterator
+from collections.abc import Iterator
+from collections.abc import Iterable
+import aiohttp
+import urllib3
+import typing
+import sys
+
+from connectrpc.client_async import AsyncConnectClient
+from connectrpc.client_sync import ConnectClient
+from connectrpc.client_protocol import ConnectProtocol
+from connectrpc.client_connect import ConnectProtocolError
+from connectrpc.headers import HeaderInput
+from connectrpc.server import ClientRequest
+from connectrpc.server import ClientStream
+from connectrpc.server import ServerResponse
+from connectrpc.server import ServerStream
+from connectrpc.server_sync import ConnectWSGI
+from connectrpc.streams import StreamInput
+from connectrpc.streams import AsyncStreamOutput
+from connectrpc.streams import StreamOutput
+from connectrpc.unary import UnaryOutput
+from connectrpc.unary import ClientStreamingOutput
+
+if typing.TYPE_CHECKING:
+    # wsgiref.types was added in Python 3.11.
+    if sys.version_info >= (3, 11):
+        from wsgiref.types import WSGIApplication
+    else:
+        from _typeshed.wsgi import WSGIApplication
+
+import authorization.v2.authorization_pb2
+
+class AuthorizationServiceClient:
+    def __init__(
+        self,
+        base_url: str,
+        http_client: urllib3.PoolManager | None = None,
+        protocol: ConnectProtocol = ConnectProtocol.CONNECT_PROTOBUF,
+    ):
+        self.base_url = base_url
+        self._connect_client = ConnectClient(http_client, protocol)
+    def call_get_decision(
+        self, req: authorization.v2.authorization_pb2.GetDecisionRequest,extra_headers: HeaderInput | None=None, timeout_seconds: float | None=None
+    ) -> UnaryOutput[authorization.v2.authorization_pb2.GetDecisionResponse]:
+        """Low-level method to call GetDecision, granting access to errors and metadata"""
+        url = self.base_url + "/authorization.v2.AuthorizationService/GetDecision"
+        return self._connect_client.call_unary(url, req, authorization.v2.authorization_pb2.GetDecisionResponse,extra_headers, timeout_seconds)
+
+
+    def get_decision(
+        self, req: authorization.v2.authorization_pb2.GetDecisionRequest,extra_headers: HeaderInput | None=None, timeout_seconds: float | None=None
+    ) -> authorization.v2.authorization_pb2.GetDecisionResponse:
+        response = self.call_get_decision(req, extra_headers, timeout_seconds)
+        err = response.error()
+        if err is not None:
+            raise err
+        msg = response.message()
+        if msg is None:
+            raise ConnectProtocolError('missing response message')
+        return msg
+
+    def call_get_decision_multi_resource(
+        self, req: authorization.v2.authorization_pb2.GetDecisionMultiResourceRequest,extra_headers: HeaderInput | None=None, timeout_seconds: float | None=None
+    ) -> UnaryOutput[authorization.v2.authorization_pb2.GetDecisionMultiResourceResponse]:
+        """Low-level method to call GetDecisionMultiResource, granting access to errors and metadata"""
+        url = self.base_url + "/authorization.v2.AuthorizationService/GetDecisionMultiResource"
+        return self._connect_client.call_unary(url, req, authorization.v2.authorization_pb2.GetDecisionMultiResourceResponse,extra_headers, timeout_seconds)
+
+
+    def get_decision_multi_resource(
+        self, req: authorization.v2.authorization_pb2.GetDecisionMultiResourceRequest,extra_headers: HeaderInput | None=None, timeout_seconds: float | None=None
+    ) -> authorization.v2.authorization_pb2.GetDecisionMultiResourceResponse:
+        response = self.call_get_decision_multi_resource(req, extra_headers, timeout_seconds)
+        err = response.error()
+        if err is not None:
+            raise err
+        msg = response.message()
+        if msg is None:
+            raise ConnectProtocolError('missing response message')
+        return msg
+
+    def call_get_decision_bulk(
+        self, req: authorization.v2.authorization_pb2.GetDecisionBulkRequest,extra_headers: HeaderInput | None=None, timeout_seconds: float | None=None
+    ) -> UnaryOutput[authorization.v2.authorization_pb2.GetDecisionBulkResponse]:
+        """Low-level method to call GetDecisionBulk, granting access to errors and metadata"""
+        url = self.base_url + "/authorization.v2.AuthorizationService/GetDecisionBulk"
+        return self._connect_client.call_unary(url, req, authorization.v2.authorization_pb2.GetDecisionBulkResponse,extra_headers, timeout_seconds)
+
+
+    def get_decision_bulk(
+        self, req: authorization.v2.authorization_pb2.GetDecisionBulkRequest,extra_headers: HeaderInput | None=None, timeout_seconds: float | None=None
+    ) -> authorization.v2.authorization_pb2.GetDecisionBulkResponse:
+        response = self.call_get_decision_bulk(req, extra_headers, timeout_seconds)
+        err = response.error()
+        if err is not None:
+            raise err
+        msg = response.message()
+        if msg is None:
+            raise ConnectProtocolError('missing response message')
+        return msg
+
+    def call_get_entitlements(
+        self, req: authorization.v2.authorization_pb2.GetEntitlementsRequest,extra_headers: HeaderInput | None=None, timeout_seconds: float | None=None
+    ) -> UnaryOutput[authorization.v2.authorization_pb2.GetEntitlementsResponse]:
+        """Low-level method to call GetEntitlements, granting access to errors and metadata"""
+        url = self.base_url + "/authorization.v2.AuthorizationService/GetEntitlements"
+        return self._connect_client.call_unary(url, req, authorization.v2.authorization_pb2.GetEntitlementsResponse,extra_headers, timeout_seconds)
+
+
+    def get_entitlements(
+        self, req: authorization.v2.authorization_pb2.GetEntitlementsRequest,extra_headers: HeaderInput | None=None, timeout_seconds: float | None=None
+    ) -> authorization.v2.authorization_pb2.GetEntitlementsResponse:
+        response = self.call_get_entitlements(req, extra_headers, timeout_seconds)
+        err = response.error()
+        if err is not None:
+            raise err
+        msg = response.message()
+        if msg is None:
+            raise ConnectProtocolError('missing response message')
+        return msg
+
+
+class AsyncAuthorizationServiceClient:
+    def __init__(
+        self,
+        base_url: str,
+        http_client: aiohttp.ClientSession,
+        protocol: ConnectProtocol = ConnectProtocol.CONNECT_PROTOBUF,
+    ):
+        self.base_url = base_url
+        self._connect_client = AsyncConnectClient(http_client, protocol)
+
+    async def call_get_decision(
+        self, req: authorization.v2.authorization_pb2.GetDecisionRequest,extra_headers: HeaderInput | None=None, timeout_seconds: float | None=None
+    ) -> UnaryOutput[authorization.v2.authorization_pb2.GetDecisionResponse]:
+        """Low-level method to call GetDecision, granting access to errors and metadata"""
+        url = self.base_url + "/authorization.v2.AuthorizationService/GetDecision"
+        return await self._connect_client.call_unary(url, req, authorization.v2.authorization_pb2.GetDecisionResponse,extra_headers, timeout_seconds)
+
+    async def get_decision(
+        self, req: authorization.v2.authorization_pb2.GetDecisionRequest,extra_headers: HeaderInput | None=None, timeout_seconds: float | None=None
+    ) -> authorization.v2.authorization_pb2.GetDecisionResponse:
+        response = await self.call_get_decision(req, extra_headers, timeout_seconds)
+        err = response.error()
+        if err is not None:
+            raise err
+        msg = response.message()
+        if msg is None:
+            raise ConnectProtocolError('missing response message')
+        return msg
+
+    async def call_get_decision_multi_resource(
+        self, req: authorization.v2.authorization_pb2.GetDecisionMultiResourceRequest,extra_headers: HeaderInput | None=None, timeout_seconds: float | None=None
+    ) -> UnaryOutput[authorization.v2.authorization_pb2.GetDecisionMultiResourceResponse]:
+        """Low-level method to call GetDecisionMultiResource, granting access to errors and metadata"""
+        url = self.base_url + "/authorization.v2.AuthorizationService/GetDecisionMultiResource"
+        return await self._connect_client.call_unary(url, req, authorization.v2.authorization_pb2.GetDecisionMultiResourceResponse,extra_headers, timeout_seconds)
+
+    async def get_decision_multi_resource(
+        self, req: authorization.v2.authorization_pb2.GetDecisionMultiResourceRequest,extra_headers: HeaderInput | None=None, timeout_seconds: float | None=None
+    ) -> authorization.v2.authorization_pb2.GetDecisionMultiResourceResponse:
+        response = await self.call_get_decision_multi_resource(req, extra_headers, timeout_seconds)
+        err = response.error()
+        if err is not None:
+            raise err
+        msg = response.message()
+        if msg is None:
+            raise ConnectProtocolError('missing response message')
+        return msg
+
+    async def call_get_decision_bulk(
+        self, req: authorization.v2.authorization_pb2.GetDecisionBulkRequest,extra_headers: HeaderInput | None=None, timeout_seconds: float | None=None
+    ) -> UnaryOutput[authorization.v2.authorization_pb2.GetDecisionBulkResponse]:
+        """Low-level method to call GetDecisionBulk, granting access to errors and metadata"""
+        url = self.base_url + "/authorization.v2.AuthorizationService/GetDecisionBulk"
+        return await self._connect_client.call_unary(url, req, authorization.v2.authorization_pb2.GetDecisionBulkResponse,extra_headers, timeout_seconds)
+
+    async def get_decision_bulk(
+        self, req: authorization.v2.authorization_pb2.GetDecisionBulkRequest,extra_headers: HeaderInput | None=None, timeout_seconds: float | None=None
+    ) -> authorization.v2.authorization_pb2.GetDecisionBulkResponse:
+        response = await self.call_get_decision_bulk(req, extra_headers, timeout_seconds)
+        err = response.error()
+        if err is not None:
+            raise err
+        msg = response.message()
+        if msg is None:
+            raise ConnectProtocolError('missing response message')
+        return msg
+
+    async def call_get_entitlements(
+        self, req: authorization.v2.authorization_pb2.GetEntitlementsRequest,extra_headers: HeaderInput | None=None, timeout_seconds: float | None=None
+    ) -> UnaryOutput[authorization.v2.authorization_pb2.GetEntitlementsResponse]:
+        """Low-level method to call GetEntitlements, granting access to errors and metadata"""
+        url = self.base_url + "/authorization.v2.AuthorizationService/GetEntitlements"
+        return await self._connect_client.call_unary(url, req, authorization.v2.authorization_pb2.GetEntitlementsResponse,extra_headers, timeout_seconds)
+
+    async def get_entitlements(
+        self, req: authorization.v2.authorization_pb2.GetEntitlementsRequest,extra_headers: HeaderInput | None=None, timeout_seconds: float | None=None
+    ) -> authorization.v2.authorization_pb2.GetEntitlementsResponse:
+        response = await self.call_get_entitlements(req, extra_headers, timeout_seconds)
+        err = response.error()
+        if err is not None:
+            raise err
+        msg = response.message()
+        if msg is None:
+            raise ConnectProtocolError('missing response message')
+        return msg
+
+
+@typing.runtime_checkable
+class AuthorizationServiceProtocol(typing.Protocol):
+    def get_decision(self, req: ClientRequest[authorization.v2.authorization_pb2.GetDecisionRequest]) -> ServerResponse[authorization.v2.authorization_pb2.GetDecisionResponse]:
+        ...
+    def get_decision_multi_resource(self, req: ClientRequest[authorization.v2.authorization_pb2.GetDecisionMultiResourceRequest]) -> ServerResponse[authorization.v2.authorization_pb2.GetDecisionMultiResourceResponse]:
+        ...
+    def get_decision_bulk(self, req: ClientRequest[authorization.v2.authorization_pb2.GetDecisionBulkRequest]) -> ServerResponse[authorization.v2.authorization_pb2.GetDecisionBulkResponse]:
+        ...
+    def get_entitlements(self, req: ClientRequest[authorization.v2.authorization_pb2.GetEntitlementsRequest]) -> ServerResponse[authorization.v2.authorization_pb2.GetEntitlementsResponse]:
+        ...
+
+AUTHORIZATION_SERVICE_PATH_PREFIX = "/authorization.v2.AuthorizationService"
+
+def wsgi_authorization_service(implementation: AuthorizationServiceProtocol) -> WSGIApplication:
+    app = ConnectWSGI()
+    app.register_unary_rpc("/authorization.v2.AuthorizationService/GetDecision", implementation.get_decision, authorization.v2.authorization_pb2.GetDecisionRequest)
+    app.register_unary_rpc("/authorization.v2.AuthorizationService/GetDecisionMultiResource", implementation.get_decision_multi_resource, authorization.v2.authorization_pb2.GetDecisionMultiResourceRequest)
+    app.register_unary_rpc("/authorization.v2.AuthorizationService/GetDecisionBulk", implementation.get_decision_bulk, authorization.v2.authorization_pb2.GetDecisionBulkRequest)
+    app.register_unary_rpc("/authorization.v2.AuthorizationService/GetEntitlements", implementation.get_entitlements, authorization.v2.authorization_pb2.GetEntitlementsRequest)
+    return app

otdf_python_proto/common/__init__.py

@@ -0,0 +1 @@
+"""common protobuf definitions."""

otdf_python_proto/common/common_pb2.py

@@ -0,0 +1,52 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: common/common.proto
+# Protobuf Python Version: 6.31.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    6,
+    31,
+    1,
+    '',
+    'common/common.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x13\x63ommon/common.proto\x12\x06\x63ommon\x1a\x1fgoogle/protobuf/timestamp.proto\"\xf1\x01\n\x08Metadata\x12\x39\n\ncreated_at\x18\x01 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAt\x12\x39\n\nupdated_at\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tupdatedAt\x12\x34\n\x06labels\x18\x03 \x03(\x0b\x32\x1c.common.Metadata.LabelsEntryR\x06labels\x1a\x39\n\x0bLabelsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01\"\x89\x01\n\x0fMetadataMutable\x12;\n\x06labels\x18\x03 \x03(\x0b\x32#.common.MetadataMutable.LabelsEntryR\x06labels\x1a\x39\n\x0bLabelsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01*}\n\x12MetadataUpdateEnum\x12$\n METADATA_UPDATE_ENUM_UNSPECIFIED\x10\x00\x12\x1f\n\x1bMETADATA_UPDATE_ENUM_EXTEND\x10\x01\x12 \n\x1cMETADATA_UPDATE_ENUM_REPLACE\x10\x02*\x8d\x01\n\x0f\x41\x63tiveStateEnum\x12!\n\x1d\x41\x43TIVE_STATE_ENUM_UNSPECIFIED\x10\x00\x12\x1c\n\x18\x41\x43TIVE_STATE_ENUM_ACTIVE\x10\x01\x12\x1e\n\x1a\x41\x43TIVE_STATE_ENUM_INACTIVE\x10\x02\x12\x19\n\x15\x41\x43TIVE_STATE_ENUM_ANY\x10\x03\x42Q\n\ncom.commonB\x0b\x43ommonProtoP\x01\xa2\x02\x03\x43XX\xaa\x02\x06\x43ommon\xca\x02\x06\x43ommon\xe2\x02\x12\x43ommon\\GPBMetadata\xea\x02\x06\x43ommonb\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'common.common_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  _globals['DESCRIPTOR']._loaded_options = None
+  _globals['DESCRIPTOR']._serialized_options = b'\n\ncom.commonB\013CommonProtoP\001\242\002\003CXX\252\002\006Common\312\002\006Common\342\002\022Common\\GPBMetadata\352\002\006Common'
+  _globals['_METADATA_LABELSENTRY']._loaded_options = None
+  _globals['_METADATA_LABELSENTRY']._serialized_options = b'8\001'
+  _globals['_METADATAMUTABLE_LABELSENTRY']._loaded_options = None
+  _globals['_METADATAMUTABLE_LABELSENTRY']._serialized_options = b'8\001'
+  _globals['_METADATAUPDATEENUM']._serialized_start=448
+  _globals['_METADATAUPDATEENUM']._serialized_end=573
+  _globals['_ACTIVESTATEENUM']._serialized_start=576
+  _globals['_ACTIVESTATEENUM']._serialized_end=717
+  _globals['_METADATA']._serialized_start=65
+  _globals['_METADATA']._serialized_end=306
+  _globals['_METADATA_LABELSENTRY']._serialized_start=249
+  _globals['_METADATA_LABELSENTRY']._serialized_end=306
+  _globals['_METADATAMUTABLE']._serialized_start=309
+  _globals['_METADATAMUTABLE']._serialized_end=446
+  _globals['_METADATAMUTABLE_LABELSENTRY']._serialized_start=249
+  _globals['_METADATAMUTABLE_LABELSENTRY']._serialized_end=306
+# @@protoc_insertion_point(module_scope)

otdf_python_proto/common/common_pb2.pyi

@@ -0,0 +1,61 @@
+import datetime
+
+from google.protobuf import timestamp_pb2 as _timestamp_pb2
+from google.protobuf.internal import containers as _containers
+from google.protobuf.internal import enum_type_wrapper as _enum_type_wrapper
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import message as _message
+from collections.abc import Mapping as _Mapping
+from typing import ClassVar as _ClassVar, Optional as _Optional, Union as _Union
+
+DESCRIPTOR: _descriptor.FileDescriptor
+
+class MetadataUpdateEnum(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
+    __slots__ = ()
+    METADATA_UPDATE_ENUM_UNSPECIFIED: _ClassVar[MetadataUpdateEnum]
+    METADATA_UPDATE_ENUM_EXTEND: _ClassVar[MetadataUpdateEnum]
+    METADATA_UPDATE_ENUM_REPLACE: _ClassVar[MetadataUpdateEnum]
+
+class ActiveStateEnum(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
+    __slots__ = ()
+    ACTIVE_STATE_ENUM_UNSPECIFIED: _ClassVar[ActiveStateEnum]
+    ACTIVE_STATE_ENUM_ACTIVE: _ClassVar[ActiveStateEnum]
+    ACTIVE_STATE_ENUM_INACTIVE: _ClassVar[ActiveStateEnum]
+    ACTIVE_STATE_ENUM_ANY: _ClassVar[ActiveStateEnum]
+METADATA_UPDATE_ENUM_UNSPECIFIED: MetadataUpdateEnum
+METADATA_UPDATE_ENUM_EXTEND: MetadataUpdateEnum
+METADATA_UPDATE_ENUM_REPLACE: MetadataUpdateEnum
+ACTIVE_STATE_ENUM_UNSPECIFIED: ActiveStateEnum
+ACTIVE_STATE_ENUM_ACTIVE: ActiveStateEnum
+ACTIVE_STATE_ENUM_INACTIVE: ActiveStateEnum
+ACTIVE_STATE_ENUM_ANY: ActiveStateEnum
+
+class Metadata(_message.Message):
+    __slots__ = ("created_at", "updated_at", "labels")
+    class LabelsEntry(_message.Message):
+        __slots__ = ("key", "value")
+        KEY_FIELD_NUMBER: _ClassVar[int]
+        VALUE_FIELD_NUMBER: _ClassVar[int]
+        key: str
+        value: str
+        def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ...
+    CREATED_AT_FIELD_NUMBER: _ClassVar[int]
+    UPDATED_AT_FIELD_NUMBER: _ClassVar[int]
+    LABELS_FIELD_NUMBER: _ClassVar[int]
+    created_at: _timestamp_pb2.Timestamp
+    updated_at: _timestamp_pb2.Timestamp
+    labels: _containers.ScalarMap[str, str]
+    def __init__(self, created_at: _Optional[_Union[datetime.datetime, _timestamp_pb2.Timestamp, _Mapping]] = ..., updated_at: _Optional[_Union[datetime.datetime, _timestamp_pb2.Timestamp, _Mapping]] = ..., labels: _Optional[_Mapping[str, str]] = ...) -> None: ...
+
+class MetadataMutable(_message.Message):
+    __slots__ = ("labels",)
+    class LabelsEntry(_message.Message):
+        __slots__ = ("key", "value")
+        KEY_FIELD_NUMBER: _ClassVar[int]
+        VALUE_FIELD_NUMBER: _ClassVar[int]
+        key: str
+        value: str
+        def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ...
+    LABELS_FIELD_NUMBER: _ClassVar[int]
+    labels: _containers.ScalarMap[str, str]
+    def __init__(self, labels: _Optional[_Mapping[str, str]] = ...) -> None: ...

otdf_python_proto/entity/__init__.py

@@ -0,0 +1 @@
+"""entity protobuf definitions."""

otdf_python_proto/entity/entity_pb2.py

@@ -0,0 +1,47 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: entity/entity.proto
+# Protobuf Python Version: 6.31.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    6,
+    31,
+    1,
+    '',
+    'entity/entity.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from buf.validate import validate_pb2 as buf_dot_validate_dot_validate__pb2
+from google.protobuf import any_pb2 as google_dot_protobuf_dot_any__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x13\x65ntity/entity.proto\x12\x06\x65ntity\x1a\x1b\x62uf/validate/validate.proto\x1a\x19google/protobuf/any.proto\"D\n\x05Token\x12!\n\x0c\x65phemeral_id\x18\x01 \x01(\tR\x0b\x65phemeralId\x12\x18\n\x03jwt\x18\x02 \x01(\tB\x06\xbaH\x03\xc8\x01\x01R\x03jwt\"\xda\x02\n\x06\x45ntity\x12!\n\x0c\x65phemeral_id\x18\x01 \x01(\tR\x0b\x65phemeralId\x12%\n\remail_address\x18\x02 \x01(\tH\x00R\x0c\x65mailAddress\x12\x1d\n\tuser_name\x18\x03 \x01(\tH\x00R\x08userName\x12.\n\x06\x63laims\x18\x04 \x01(\x0b\x32\x14.google.protobuf.AnyH\x00R\x06\x63laims\x12\x1d\n\tclient_id\x18\x05 \x01(\tH\x00R\x08\x63lientId\x12\x33\n\x08\x63\x61tegory\x18\x0b \x01(\x0e\x32\x17.entity.Entity.CategoryR\x08\x63\x61tegory\"T\n\x08\x43\x61tegory\x12\x18\n\x14\x43\x41TEGORY_UNSPECIFIED\x10\x00\x12\x14\n\x10\x43\x41TEGORY_SUBJECT\x10\x01\x12\x18\n\x14\x43\x41TEGORY_ENVIRONMENT\x10\x02\x42\r\n\x0b\x65ntity_type\"\\\n\x0b\x45ntityChain\x12!\n\x0c\x65phemeral_id\x18\x01 \x01(\tR\x0b\x65phemeralId\x12*\n\x08\x65ntities\x18\x02 \x03(\x0b\x32\x0e.entity.EntityR\x08\x65ntitiesBQ\n\ncom.entityB\x0b\x45ntityProtoP\x01\xa2\x02\x03\x45XX\xaa\x02\x06\x45ntity\xca\x02\x06\x45ntity\xe2\x02\x12\x45ntity\\GPBMetadata\xea\x02\x06\x45ntityb\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'entity.entity_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  _globals['DESCRIPTOR']._loaded_options = None
+  _globals['DESCRIPTOR']._serialized_options = b'\n\ncom.entityB\013EntityProtoP\001\242\002\003EXX\252\002\006Entity\312\002\006Entity\342\002\022Entity\\GPBMetadata\352\002\006Entity'
+  _globals['_TOKEN'].fields_by_name['jwt']._loaded_options = None
+  _globals['_TOKEN'].fields_by_name['jwt']._serialized_options = b'\272H\003\310\001\001'
+  _globals['_TOKEN']._serialized_start=87
+  _globals['_TOKEN']._serialized_end=155
+  _globals['_ENTITY']._serialized_start=158
+  _globals['_ENTITY']._serialized_end=504
+  _globals['_ENTITY_CATEGORY']._serialized_start=405
+  _globals['_ENTITY_CATEGORY']._serialized_end=489
+  _globals['_ENTITYCHAIN']._serialized_start=506
+  _globals['_ENTITYCHAIN']._serialized_end=598
+# @@protoc_insertion_point(module_scope)

otdf_python_proto/entity/entity_pb2.pyi

@@ -0,0 +1,50 @@
+from buf.validate import validate_pb2 as _validate_pb2
+from google.protobuf import any_pb2 as _any_pb2
+from google.protobuf.internal import containers as _containers
+from google.protobuf.internal import enum_type_wrapper as _enum_type_wrapper
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import message as _message
+from collections.abc import Iterable as _Iterable, Mapping as _Mapping
+from typing import ClassVar as _ClassVar, Optional as _Optional, Union as _Union
+
+DESCRIPTOR: _descriptor.FileDescriptor
+
+class Token(_message.Message):
+    __slots__ = ("ephemeral_id", "jwt")
+    EPHEMERAL_ID_FIELD_NUMBER: _ClassVar[int]
+    JWT_FIELD_NUMBER: _ClassVar[int]
+    ephemeral_id: str
+    jwt: str
+    def __init__(self, ephemeral_id: _Optional[str] = ..., jwt: _Optional[str] = ...) -> None: ...
+
+class Entity(_message.Message):
+    __slots__ = ("ephemeral_id", "email_address", "user_name", "claims", "client_id", "category")
+    class Category(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
+        __slots__ = ()
+        CATEGORY_UNSPECIFIED: _ClassVar[Entity.Category]
+        CATEGORY_SUBJECT: _ClassVar[Entity.Category]
+        CATEGORY_ENVIRONMENT: _ClassVar[Entity.Category]
+    CATEGORY_UNSPECIFIED: Entity.Category
+    CATEGORY_SUBJECT: Entity.Category
+    CATEGORY_ENVIRONMENT: Entity.Category
+    EPHEMERAL_ID_FIELD_NUMBER: _ClassVar[int]
+    EMAIL_ADDRESS_FIELD_NUMBER: _ClassVar[int]
+    USER_NAME_FIELD_NUMBER: _ClassVar[int]
+    CLAIMS_FIELD_NUMBER: _ClassVar[int]
+    CLIENT_ID_FIELD_NUMBER: _ClassVar[int]
+    CATEGORY_FIELD_NUMBER: _ClassVar[int]
+    ephemeral_id: str
+    email_address: str
+    user_name: str
+    claims: _any_pb2.Any
+    client_id: str
+    category: Entity.Category
+    def __init__(self, ephemeral_id: _Optional[str] = ..., email_address: _Optional[str] = ..., user_name: _Optional[str] = ..., claims: _Optional[_Union[_any_pb2.Any, _Mapping]] = ..., client_id: _Optional[str] = ..., category: _Optional[_Union[Entity.Category, str]] = ...) -> None: ...
+
+class EntityChain(_message.Message):
+    __slots__ = ("ephemeral_id", "entities")
+    EPHEMERAL_ID_FIELD_NUMBER: _ClassVar[int]
+    ENTITIES_FIELD_NUMBER: _ClassVar[int]
+    ephemeral_id: str
+    entities: _containers.RepeatedCompositeFieldContainer[Entity]
+    def __init__(self, ephemeral_id: _Optional[str] = ..., entities: _Optional[_Iterable[_Union[Entity, _Mapping]]] = ...) -> None: ...

otdf_python_proto/entityresolution/__init__.py

@@ -0,0 +1 @@
+"""entityresolution protobuf definitions."""

otdf_python_proto/entityresolution/entity_resolution_pb2.py

@@ -0,0 +1,57 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: entityresolution/entity_resolution.proto
+# Protobuf Python Version: 6.31.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    6,
+    31,
+    1,
+    '',
+    'entityresolution/entity_resolution.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from authorization import authorization_pb2 as authorization_dot_authorization__pb2
+from google.protobuf import struct_pb2 as google_dot_protobuf_dot_struct__pb2
+from google.protobuf import any_pb2 as google_dot_protobuf_dot_any__pb2
+from google.api import annotations_pb2 as google_dot_api_dot_annotations__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n(entityresolution/entity_resolution.proto\x12\x10\x65ntityresolution\x1a!authorization/authorization.proto\x1a\x1cgoogle/protobuf/struct.proto\x1a\x19google/protobuf/any.proto\x1a\x1cgoogle/api/annotations.proto\"K\n\x16ResolveEntitiesRequest\x12\x31\n\x08\x65ntities\x18\x01 \x03(\x0b\x32\x15.authorization.EntityR\x08\x65ntities\"{\n\x14\x45ntityRepresentation\x12\x42\n\x10\x61\x64\x64itional_props\x18\x01 \x03(\x0b\x32\x17.google.protobuf.StructR\x0f\x61\x64\x64itionalProps\x12\x1f\n\x0boriginal_id\x18\x02 \x01(\tR\noriginalId\"x\n\x17ResolveEntitiesResponse\x12]\n\x16\x65ntity_representations\x18\x01 \x03(\x0b\x32&.entityresolution.EntityRepresentationR\x15\x65ntityRepresentations\"\x8b\x01\n\x13\x45ntityNotFoundError\x12\x12\n\x04\x63ode\x18\x01 \x01(\x05R\x04\x63ode\x12\x18\n\x07message\x18\x02 \x01(\tR\x07message\x12.\n\x07\x64\x65tails\x18\x03 \x03(\x0b\x32\x14.google.protobuf.AnyR\x07\x64\x65tails\x12\x16\n\x06\x65ntity\x18\x04 \x01(\tR\x06\x65ntity\"O\n\x1f\x43reateEntityChainFromJwtRequest\x12,\n\x06tokens\x18\x01 \x03(\x0b\x32\x14.authorization.TokenR\x06tokens\"c\n CreateEntityChainFromJwtResponse\x12?\n\rentity_chains\x18\x01 \x03(\x0b\x32\x1a.authorization.EntityChainR\x0c\x65ntityChains2\xd6\x02\n\x17\x45ntityResolutionService\x12\x8c\x01\n\x0fResolveEntities\x12(.entityresolution.ResolveEntitiesRequest\x1a).entityresolution.ResolveEntitiesResponse\"$\x82\xd3\xe4\x93\x02\x1e\"\x19/entityresolution/resolve:\x01*\x12\xab\x01\n\x18\x43reateEntityChainFromJwt\x12\x31.entityresolution.CreateEntityChainFromJwtRequest\x1a\x32.entityresolution.CreateEntityChainFromJwtResponse\"(\x82\xd3\xe4\x93\x02\"\"\x1d/entityresolution/entitychain:\x01*B\x8d\x01\n\x14\x63om.entityresolutionB\x15\x45ntityResolutionProtoP\x01\xa2\x02\x03\x45XX\xaa\x02\x10\x45ntityresolution\xca\x02\x10\x45ntityresolution\xe2\x02\x1c\x45ntityresolution\\GPBMetadata\xea\x02\x10\x45ntityresolutionb\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'entityresolution.entity_resolution_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  _globals['DESCRIPTOR']._loaded_options = None
+  _globals['DESCRIPTOR']._serialized_options = b'\n\024com.entityresolutionB\025EntityResolutionProtoP\001\242\002\003EXX\252\002\020Entityresolution\312\002\020Entityresolution\342\002\034Entityresolution\\GPBMetadata\352\002\020Entityresolution'
+  _globals['_ENTITYRESOLUTIONSERVICE'].methods_by_name['ResolveEntities']._loaded_options = None
+  _globals['_ENTITYRESOLUTIONSERVICE'].methods_by_name['ResolveEntities']._serialized_options = b'\202\323\344\223\002\036\"\031/entityresolution/resolve:\001*'
+  _globals['_ENTITYRESOLUTIONSERVICE'].methods_by_name['CreateEntityChainFromJwt']._loaded_options = None
+  _globals['_ENTITYRESOLUTIONSERVICE'].methods_by_name['CreateEntityChainFromJwt']._serialized_options = b'\202\323\344\223\002\"\"\035/entityresolution/entitychain:\001*'
+  _globals['_RESOLVEENTITIESREQUEST']._serialized_start=184
+  _globals['_RESOLVEENTITIESREQUEST']._serialized_end=259
+  _globals['_ENTITYREPRESENTATION']._serialized_start=261
+  _globals['_ENTITYREPRESENTATION']._serialized_end=384
+  _globals['_RESOLVEENTITIESRESPONSE']._serialized_start=386
+  _globals['_RESOLVEENTITIESRESPONSE']._serialized_end=506
+  _globals['_ENTITYNOTFOUNDERROR']._serialized_start=509
+  _globals['_ENTITYNOTFOUNDERROR']._serialized_end=648
+  _globals['_CREATEENTITYCHAINFROMJWTREQUEST']._serialized_start=650
+  _globals['_CREATEENTITYCHAINFROMJWTREQUEST']._serialized_end=729
+  _globals['_CREATEENTITYCHAINFROMJWTRESPONSE']._serialized_start=731
+  _globals['_CREATEENTITYCHAINFROMJWTRESPONSE']._serialized_end=830
+  _globals['_ENTITYRESOLUTIONSERVICE']._serialized_start=833
+  _globals['_ENTITYRESOLUTIONSERVICE']._serialized_end=1175
+# @@protoc_insertion_point(module_scope)

otdf_python_proto/entityresolution/entity_resolution_pb2.pyi

@@ -0,0 +1,55 @@
+from authorization import authorization_pb2 as _authorization_pb2
+from google.protobuf import struct_pb2 as _struct_pb2
+from google.protobuf import any_pb2 as _any_pb2
+from google.api import annotations_pb2 as _annotations_pb2
+from google.protobuf.internal import containers as _containers
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import message as _message
+from collections.abc import Iterable as _Iterable, Mapping as _Mapping
+from typing import ClassVar as _ClassVar, Optional as _Optional, Union as _Union
+
+DESCRIPTOR: _descriptor.FileDescriptor
+
+class ResolveEntitiesRequest(_message.Message):
+    __slots__ = ("entities",)
+    ENTITIES_FIELD_NUMBER: _ClassVar[int]
+    entities: _containers.RepeatedCompositeFieldContainer[_authorization_pb2.Entity]
+    def __init__(self, entities: _Optional[_Iterable[_Union[_authorization_pb2.Entity, _Mapping]]] = ...) -> None: ...
+
+class EntityRepresentation(_message.Message):
+    __slots__ = ("additional_props", "original_id")
+    ADDITIONAL_PROPS_FIELD_NUMBER: _ClassVar[int]
+    ORIGINAL_ID_FIELD_NUMBER: _ClassVar[int]
+    additional_props: _containers.RepeatedCompositeFieldContainer[_struct_pb2.Struct]
+    original_id: str
+    def __init__(self, additional_props: _Optional[_Iterable[_Union[_struct_pb2.Struct, _Mapping]]] = ..., original_id: _Optional[str] = ...) -> None: ...
+
+class ResolveEntitiesResponse(_message.Message):
+    __slots__ = ("entity_representations",)
+    ENTITY_REPRESENTATIONS_FIELD_NUMBER: _ClassVar[int]
+    entity_representations: _containers.RepeatedCompositeFieldContainer[EntityRepresentation]
+    def __init__(self, entity_representations: _Optional[_Iterable[_Union[EntityRepresentation, _Mapping]]] = ...) -> None: ...
+
+class EntityNotFoundError(_message.Message):
+    __slots__ = ("code", "message", "details", "entity")
+    CODE_FIELD_NUMBER: _ClassVar[int]
+    MESSAGE_FIELD_NUMBER: _ClassVar[int]
+    DETAILS_FIELD_NUMBER: _ClassVar[int]
+    ENTITY_FIELD_NUMBER: _ClassVar[int]
+    code: int
+    message: str
+    details: _containers.RepeatedCompositeFieldContainer[_any_pb2.Any]
+    entity: str
+    def __init__(self, code: _Optional[int] = ..., message: _Optional[str] = ..., details: _Optional[_Iterable[_Union[_any_pb2.Any, _Mapping]]] = ..., entity: _Optional[str] = ...) -> None: ...
+
+class CreateEntityChainFromJwtRequest(_message.Message):
+    __slots__ = ("tokens",)
+    TOKENS_FIELD_NUMBER: _ClassVar[int]
+    tokens: _containers.RepeatedCompositeFieldContainer[_authorization_pb2.Token]
+    def __init__(self, tokens: _Optional[_Iterable[_Union[_authorization_pb2.Token, _Mapping]]] = ...) -> None: ...
+
+class CreateEntityChainFromJwtResponse(_message.Message):
+    __slots__ = ("entity_chains",)
+    ENTITY_CHAINS_FIELD_NUMBER: _ClassVar[int]
+    entity_chains: _containers.RepeatedCompositeFieldContainer[_authorization_pb2.EntityChain]
+    def __init__(self, entity_chains: _Optional[_Iterable[_Union[_authorization_pb2.EntityChain, _Mapping]]] = ...) -> None: ...