Merge pull request #170 from mateenkasim/improve-pyi

Include re-exports in pyi file

Author: tstirrat15

authzed/api/v1/__init__.pyi

@@ -1,15 +1,72 @@
-from authzed.api.v1.schema_service_pb2_grpc import SchemaServiceStub, SchemaServiceAsyncStub
+from typing import Any, Optional, Sequence, Tuple
+
+import grpc
+
+from authzed.api.v1.core_pb2 import (
+    AlgebraicSubjectSet,
+    ContextualizedCaveat,
+    DirectSubjectSet,
+    ObjectReference,
+    PermissionRelationshipTree,
+    Relationship,
+    RelationshipUpdate,
+    SubjectReference,
+    ZedToken,
+)
+from authzed.api.v1.error_reason_pb2 import ErrorReason
+from authzed.api.v1.experimental_service_pb2 import (
+    BulkCheckPermissionPair,
+    BulkCheckPermissionRequest,
+    BulkCheckPermissionRequestItem,
+    BulkCheckPermissionResponse,
+    BulkCheckPermissionResponseItem,
+    BulkExportRelationshipsRequest,
+    BulkExportRelationshipsResponse,
+    BulkImportRelationshipsRequest,
+    BulkImportRelationshipsResponse,
+)
 from authzed.api.v1.experimental_service_pb2_grpc import (
-    ExperimentalServiceStub,
     ExperimentalServiceAsyncStub,
+    ExperimentalServiceStub,
+)
+from authzed.api.v1.permission_service_pb2 import (
+    CheckBulkPermissionsPair,
+    CheckBulkPermissionsRequest,
+    CheckBulkPermissionsRequestItem,
+    CheckBulkPermissionsResponse,
+    CheckBulkPermissionsResponseItem,
+    CheckPermissionRequest,
+    CheckPermissionResponse,
+    Consistency,
+    DeleteRelationshipsRequest,
+    DeleteRelationshipsResponse,
+    ExpandPermissionTreeRequest,
+    ExpandPermissionTreeResponse,
+    LookupResourcesRequest,
+    LookupResourcesResponse,
+    LookupSubjectsRequest,
+    LookupSubjectsResponse,
+    Precondition,
+    ReadRelationshipsRequest,
+    ReadRelationshipsResponse,
+    RelationshipFilter,
+    SubjectFilter,
+    WriteRelationshipsRequest,
+    WriteRelationshipsResponse,
 )
 from authzed.api.v1.permission_service_pb2_grpc import (
-    PermissionsServiceStub,
     PermissionsServiceAsyncStub,
+    PermissionsServiceStub,
 )
-from authzed.api.v1.watch_service_pb2_grpc import WatchServiceStub, WatchServiceAsyncStub
-import grpc
-from typing import Optional, Sequence, Tuple, Any
+from authzed.api.v1.schema_service_pb2 import (
+    ReadSchemaRequest,
+    ReadSchemaResponse,
+    WriteSchemaRequest,
+    WriteSchemaResponse,
+)
+from authzed.api.v1.schema_service_pb2_grpc import SchemaServiceAsyncStub, SchemaServiceStub
+from authzed.api.v1.watch_service_pb2 import WatchRequest, WatchResponse
+from authzed.api.v1.watch_service_pb2_grpc import WatchServiceAsyncStub, WatchServiceStub
 
 class Client(SchemaServiceStub, PermissionsServiceStub, ExperimentalServiceStub, WatchServiceStub):
     """The Client is typed as a synchronous client (though in practice it works with both sync and async code).
@@ -48,3 +105,61 @@ class AsyncClient(
         options: Optional[Sequence[Tuple[str, Any]]] = None,
         compression: Optional[grpc.Compression] = None,
     ) -> None: ...
+
+__all__ = [
+    "Client",
+    # Core
+    "AlgebraicSubjectSet",
+    "ContextualizedCaveat",
+    "DirectSubjectSet",
+    "ObjectReference",
+    "PermissionRelationshipTree",
+    "Relationship",
+    "RelationshipUpdate",
+    "SubjectReference",
+    "ZedToken",
+    # Error Reason
+    "ErrorReason",
+    # Permission Service
+    "CheckBulkPermissionsPair",
+    "CheckBulkPermissionsRequest",
+    "CheckBulkPermissionsRequestItem",
+    "CheckBulkPermissionsResponse",
+    "CheckBulkPermissionsResponseItem",
+    "CheckPermissionRequest",
+    "CheckPermissionResponse",
+    "Consistency",
+    "DeleteRelationshipsRequest",
+    "DeleteRelationshipsResponse",
+    "ExpandPermissionTreeRequest",
+    "ExpandPermissionTreeResponse",
+    "LookupResourcesRequest",
+    "LookupResourcesResponse",
+    "LookupSubjectsRequest",
+    "LookupSubjectsResponse",
+    "Precondition",
+    "ReadRelationshipsRequest",
+    "ReadRelationshipsResponse",
+    "RelationshipFilter",
+    "SubjectFilter",
+    "WriteRelationshipsRequest",
+    "WriteRelationshipsResponse",
+    # Schema Service
+    "ReadSchemaRequest",
+    "ReadSchemaResponse",
+    "WriteSchemaRequest",
+    "WriteSchemaResponse",
+    # Watch Service
+    "WatchRequest",
+    "WatchResponse",
+    # Experimental Service
+    "BulkCheckPermissionRequest",
+    "BulkCheckPermissionResponse",
+    "BulkCheckPermissionPair",
+    "BulkCheckPermissionRequestItem",
+    "BulkCheckPermissionResponseItem",
+    "BulkImportRelationshipsRequest",
+    "BulkImportRelationshipsResponse",
+    "BulkExportRelationshipsRequest",
+    "BulkExportRelationshipsResponse",
+]