@@ -44,11 +44,11 @@ def message_decoder(
4444 dedent (
4545 f"""\
4646 def _{ m .name } Decoder(
47- d: Mapping[str, Any],
47+ d: Mapping[str, Any],
4848 ) -> { module_name } _pb2.{ m .name } :
49- m = { module_name } _pb2.{ m .name } ()
50- if d is None:
51- return m
49+ m = { module_name } _pb2.{ m .name } ()
50+ if d is None:
51+ return m
5252 """
5353 ),
5454 ]
@@ -61,12 +61,12 @@ def _{m.name}Decoder(
6161 oneofs [field .oneof_index ].append (field )
6262 continue
6363 chunks .append (
64- f" if d.get({ repr (to_camel_case (field .name ))} ) is not None:" ,
64+ f" if d.get({ repr (to_camel_case (field .name ))} ) is not None:" ,
6565 )
6666 if field .type_name == ".google.protobuf.Timestamp" :
6767 chunks .append (
6868 reindent (
69- " " ,
69+ " " ,
7070 f"""\
7171 _{ field .name } = timestamp_pb2.Timestamp()
7272 _{ field .name } .FromDatetime(d[{ repr (to_camel_case (field .name ))} ])
@@ -77,7 +77,7 @@ def _{m.name}Decoder(
7777 elif field .type_name == ".google.protobuf.BoolValue" :
7878 chunks .append (
7979 reindent (
80- " " ,
80+ " " ,
8181 f"""\
8282 _{ field .name } = BoolValue()
8383 _{ field .name } .value = d[{ repr (to_camel_case (field .name ))} ]
@@ -90,7 +90,7 @@ def _{m.name}Decoder(
9090 decode_method_name = get_decoder_name (field )
9191 chunks .append (
9292 reindent (
93- " " ,
93+ " " ,
9494 f"""\
9595 m.{ field .name } .extend(
9696 { decode_method_name } (item)
@@ -102,7 +102,7 @@ def _{m.name}Decoder(
102102 else :
103103 chunks .append (
104104 reindent (
105- " " ,
105+ " " ,
106106 f"""\
107107 m.{ field .name } .MergeFrom(d[{ repr (to_camel_case (field .name ))} ])
108108 """ ,
@@ -112,7 +112,7 @@ def _{m.name}Decoder(
112112 decode_method_name = get_decoder_name (field )
113113 chunks .append (
114114 reindent (
115- " " ,
115+ " " ,
116116 f"""\
117117 m.{ field .name } .MergeFrom(
118118 { decode_method_name } (d[{ repr (to_camel_case (field .name ))} ])
@@ -123,7 +123,7 @@ def _{m.name}Decoder(
123123 else :
124124 chunks .append (
125125 reindent (
126- " " ,
126+ " " ,
127127 f"""\
128128 setattr(
129129 m,
@@ -138,7 +138,7 @@ def _{m.name}Decoder(
138138 chunks .extend (
139139 [
140140 reindent (
141- " " ,
141+ " " ,
142142 f"""\
143143 _{ oneof .name } = d.get({ repr (to_camel_case (oneof .name ))} , {{}})
144144 if _{ oneof .name } :
@@ -150,7 +150,7 @@ def _{m.name}Decoder(
150150 for field in oneofs [index ]:
151151 chunks .append (
152152 reindent (
153- " " ,
153+ " " ,
154154 f"""\
155155 case { repr (to_camel_case (field .name ))} :
156156 """ ,
@@ -159,7 +159,7 @@ def _{m.name}Decoder(
159159 if field .type_name == ".google.protobuf.Timestamp" :
160160 chunks .append (
161161 reindent (
162- " " ,
162+ " " ,
163163 f"""\
164164 _{ field .name } = timestamp_pb2.Timestamp()
165165 _{ field .name } .FromDatetime(
@@ -172,7 +172,7 @@ def _{m.name}Decoder(
172172 elif field .label == FieldDescriptor .LABEL_REPEATED :
173173 chunks .append (
174174 reindent (
175- " " ,
175+ " " ,
176176 f"""\
177177 m.{ field .name } .MergeFrom(
178178 _{ oneof .name } [{ repr (to_camel_case (field .name ))} ]
@@ -184,7 +184,7 @@ def _{m.name}Decoder(
184184 decode_method_name = get_decoder_name (field )
185185 chunks .append (
186186 reindent (
187- " " ,
187+ " " ,
188188 f"""\
189189 m.{ field .name } .MergeFrom(
190190 { decode_method_name } (_{ oneof .name } [{ repr (to_camel_case (field .name ))} ])
@@ -195,7 +195,7 @@ def _{m.name}Decoder(
195195 else :
196196 chunks .append (
197197 reindent (
198- " " ,
198+ " " ,
199199 f"""\
200200 setattr(
201201 m,
@@ -207,7 +207,7 @@ def _{m.name}Decoder(
207207 )
208208 chunks .extend (
209209 [
210- " return m" ,
210+ " return m" ,
211211 "" ,
212212 ]
213213 )
@@ -225,7 +225,7 @@ def message_encoder(
225225 def _{ m .name } Encoder(
226226 e: { module_name } _pb2.{ m .name }
227227 ) -> Dict[str, Any]:
228- d: Dict[str, Any] = {{}}
228+ d: Dict[str, Any] = {{}}
229229 """
230230 ),
231231 ]
@@ -256,17 +256,17 @@ def _{m.name}Encoder(
256256 value = f"_{ field .name } "
257257 chunks .append (
258258 reindent (
259- " " ,
259+ " " ,
260260 f"""\
261261 _{ field .name } = e.{ field .name }
262262 if _{ field .name } is not None:
263- d[{ repr (to_camel_case (field .name ))} ] = { value }
263+ d[{ repr (to_camel_case (field .name ))} ] = { value }
264264 """ ,
265265 )
266266 )
267267 # oneof fields.
268268 for index , oneof in enumerate (m .oneof_decl ):
269- chunks .append (f" match e.WhichOneof({ repr (oneof .name )} ):" )
269+ chunks .append (f" match e.WhichOneof({ repr (oneof .name )} ):" )
270270 for field in oneofs [index ]:
271271 if field .type_name == ".google.protobuf.Timestamp" :
272272 value = f"e.{ field .name } .ToDatetime()"
@@ -277,19 +277,19 @@ def _{m.name}Encoder(
277277 value = f"e.{ field .name } "
278278 chunks .append (
279279 reindent (
280- " " ,
280+ " " ,
281281 f"""\
282282 case { repr (field .name )} :
283- d[{ repr (to_camel_case (oneof .name ))} ] = {{
284- '$kind': { repr (to_camel_case (field .name ))} ,
285- { repr (to_camel_case (field .name ))} : { value } ,
286- }}
283+ d[{ repr (to_camel_case (oneof .name ))} ] = {{
284+ '$kind': { repr (to_camel_case (field .name ))} ,
285+ { repr (to_camel_case (field .name ))} : { value } ,
286+ }}
287287 """ ,
288288 )
289289 )
290290 chunks .extend (
291291 [
292- " return d" ,
292+ " return d" ,
293293 "" ,
294294 ]
295295 )
@@ -339,10 +339,10 @@ def add_{service.name}Servicer_to_server(
339339 servicer: { pb_module_name } _pb2_grpc.{ service .name } Servicer,
340340 server: river.Server,
341341 ) -> None:
342- rpc_method_handlers: Mapping[
343- Tuple[str, str],
344- Tuple[str, river.GenericRpcHandler]
345- ] = {{
342+ rpc_method_handlers: Mapping[
343+ Tuple[str, str],
344+ Tuple[str, river.GenericRpcHandler]
345+ ] = {{
346346 """
347347 ),
348348 )
@@ -372,7 +372,7 @@ def add_{service.name}Servicer_to_server(
372372
373373 chunks .append (
374374 reindent (
375- " " ,
375+ " " ,
376376 f"""\
377377 { descriptor } : (
378378 { repr (method_kind )} ,
@@ -385,8 +385,8 @@ def add_{service.name}Servicer_to_server(
385385 """ ,
386386 ),
387387 )
388- chunks .append (" }" )
389- chunks .append (" server.add_rpc_handlers(rpc_method_handlers)" )
388+ chunks .append (" }" )
389+ chunks .append (" server.add_rpc_handlers(rpc_method_handlers)" )
390390 chunks .append ("" )
391391 return chunks
392392
0 commit comments