|
1 | 1 | # Generated by the protoc-gen-connect-python. DO NOT EDIT! |
2 | 2 | # source: connectrpc/conformance/v1/conformancev1connect/service.proto |
3 | 3 | # Protobuf Python Version: (unknown) |
4 | | -# protoc-gen-connect-python version: v0.0.0-20250225131640-797060f503da+dirty |
| 4 | +# protoc-gen-connect-python version: v0.0.0-20250710124620-c0c8871def6b+dirty |
5 | 5 | """Generated connect code.""" |
6 | 6 |
|
| 7 | +import abc |
7 | 8 | from enum import Enum |
8 | 9 |
|
9 | | -from connect.client import Client |
10 | | -import connect.connect |
11 | | -from connect.handler import ClientStreamHandler, Handler, ServerStreamHandler, UnaryHandler, BidiStreamHandler |
12 | | -from connect.handler_context import HandlerContext |
13 | | -from connect.options import ClientOptions, ConnectOptions |
| 10 | +from connect import ( |
| 11 | + Client, |
| 12 | + ClientOptions, |
| 13 | + HandlerOptions, |
| 14 | + Handler, |
| 15 | + HandlerContext, |
| 16 | + IdempotencyLevel, |
| 17 | + StreamRequest, |
| 18 | + StreamResponse, |
| 19 | +) |
| 20 | +from connect import UnaryRequest as ConnectUnaryRequest |
| 21 | +from connect import UnaryResponse as ConnectUnaryResponse |
14 | 22 | from connect.connection_pool import AsyncConnectionPool |
| 23 | +from connect.handler import BidiStreamHandler, ClientStreamHandler, ServerStreamHandler, UnaryHandler |
15 | 24 | from google.protobuf.descriptor import MethodDescriptor, ServiceDescriptor |
16 | | -from connect.idempotency_level import IdempotencyLevel |
17 | 25 |
|
18 | 26 | from .. import service_pb2 |
19 | | -from ..service_pb2 import UnaryRequest, UnaryResponse, ServerStreamRequest, ServerStreamResponse, ClientStreamRequest, ClientStreamResponse, BidiStreamRequest, BidiStreamResponse, UnimplementedRequest, UnimplementedResponse, IdempotentUnaryRequest, IdempotentUnaryResponse |
| 27 | +from ..service_pb2 import ( |
| 28 | + UnaryRequest, |
| 29 | + UnaryResponse, |
| 30 | + ServerStreamRequest, |
| 31 | + ServerStreamResponse, |
| 32 | + ClientStreamRequest, |
| 33 | + ClientStreamResponse, |
| 34 | + BidiStreamRequest, |
| 35 | + BidiStreamResponse, |
| 36 | + UnimplementedRequest, |
| 37 | + UnimplementedResponse, |
| 38 | + IdempotentUnaryRequest, |
| 39 | + IdempotentUnaryResponse, |
| 40 | +) |
20 | 41 |
|
21 | 42 |
|
22 | 43 | class ConformanceServiceProcedures(Enum): |
@@ -60,33 +81,33 @@ def __init__(self, base_url: str, pool: AsyncConnectionPool, options: ClientOpti |
60 | 81 | pool, base_url + ConformanceServiceProcedures.Unimplemented.value, UnimplementedRequest, UnimplementedResponse, options |
61 | 82 | ).call_unary |
62 | 83 | self.IdempotentUnary = Client[IdempotentUnaryRequest, IdempotentUnaryResponse]( |
63 | | - pool, base_url + ConformanceServiceProcedures.IdempotentUnary.value, IdempotentUnaryRequest, IdempotentUnaryResponse, ClientOptions(idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, enable_get=True).merge(options), |
| 84 | + pool, base_url + ConformanceServiceProcedures.IdempotentUnary.value, IdempotentUnaryRequest, IdempotentUnaryResponse, ClientOptions(idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS, enable_get=True).merge(options) |
64 | 85 | ).call_unary |
65 | 86 |
|
66 | 87 |
|
67 | | -class ConformanceServiceHandler: |
| 88 | +class ConformanceServiceHandler(metaclass=abc.ABCMeta): |
68 | 89 | """Handler for the conformanceService service.""" |
69 | 90 |
|
70 | | - async def Unary(self, request: connect.connect.UnaryRequest[UnaryRequest], context: HandlerContext) -> connect.connect.UnaryResponse[UnaryResponse]: |
| 91 | + async def Unary(self, request: ConnectUnaryRequest[UnaryRequest], context: HandlerContext) -> ConnectUnaryResponse[UnaryResponse]: |
71 | 92 | raise NotImplementedError() |
72 | 93 |
|
73 | | - async def ServerStream(self, request: connect.connect.StreamRequest[ServerStreamRequest], context: HandlerContext) -> connect.connect.StreamResponse[ServerStreamResponse]: |
| 94 | + async def ServerStream(self, request: StreamRequest[ServerStreamRequest], context: HandlerContext) -> StreamResponse[ServerStreamResponse]: |
74 | 95 | raise NotImplementedError() |
75 | 96 |
|
76 | | - async def ClientStream(self, request: connect.connect.StreamRequest[ClientStreamRequest], context: HandlerContext) -> connect.connect.StreamResponse[ClientStreamResponse]: |
| 97 | + async def ClientStream(self, request: StreamRequest[ClientStreamRequest], context: HandlerContext) -> StreamResponse[ClientStreamResponse]: |
77 | 98 | raise NotImplementedError() |
78 | 99 |
|
79 | | - async def BidiStream(self, request: connect.connect.StreamRequest[BidiStreamRequest], context: HandlerContext) -> connect.connect.StreamResponse[BidiStreamResponse]: |
| 100 | + async def BidiStream(self, request: StreamRequest[BidiStreamRequest], context: HandlerContext) -> StreamResponse[BidiStreamResponse]: |
80 | 101 | raise NotImplementedError() |
81 | 102 |
|
82 | | - async def Unimplemented(self, request: connect.connect.UnaryRequest[UnimplementedRequest], context: HandlerContext) -> connect.connect.UnaryResponse[UnimplementedResponse]: |
| 103 | + async def Unimplemented(self, request: ConnectUnaryRequest[UnimplementedRequest], context: HandlerContext) -> ConnectUnaryResponse[UnimplementedResponse]: |
83 | 104 | raise NotImplementedError() |
84 | 105 |
|
85 | | - async def IdempotentUnary(self, request: connect.connect.UnaryRequest[IdempotentUnaryRequest], context: HandlerContext) -> connect.connect.UnaryResponse[IdempotentUnaryResponse]: |
| 106 | + async def IdempotentUnary(self, request: ConnectUnaryRequest[IdempotentUnaryRequest], context: HandlerContext) -> ConnectUnaryResponse[IdempotentUnaryResponse]: |
86 | 107 | raise NotImplementedError() |
87 | 108 |
|
88 | 109 |
|
89 | | -def create_ConformanceService_handlers(service: ConformanceServiceHandler, options: ConnectOptions | None = None) -> list[Handler]: |
| 110 | +def create_ConformanceService_handlers(service: ConformanceServiceHandler, options: HandlerOptions | None = None) -> list[Handler]: |
90 | 111 | handlers = [ |
91 | 112 | UnaryHandler( |
92 | 113 | procedure=ConformanceServiceProcedures.Unary.value, |
@@ -128,7 +149,7 @@ def create_ConformanceService_handlers(service: ConformanceServiceHandler, optio |
128 | 149 | unary=service.IdempotentUnary, |
129 | 150 | input=IdempotentUnaryRequest, |
130 | 151 | output=IdempotentUnaryResponse, |
131 | | - options=ConnectOptions(idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS).merge(options), |
| 152 | + options=HandlerOptions(idempotency_level=IdempotencyLevel.NO_SIDE_EFFECTS).merge(options), |
132 | 153 | ), |
133 | 154 | ] |
134 | 155 | return handlers |
0 commit comments