|
67 | 67 |
|
68 | 68 | from gapic import utils |
69 | 69 | from gapic.schema import metadata |
| 70 | +from gapic.utils import cached_proto_context |
70 | 71 | from gapic.utils import uri_sample |
71 | 72 | from gapic.utils import make_private |
72 | | -from gapic.utils import cached_proto_context |
73 | 73 |
|
74 | 74 |
|
75 | 75 | @dataclasses.dataclass(frozen=True) |
@@ -437,10 +437,8 @@ def with_context( |
437 | 437 | if self.message |
438 | 438 | else None |
439 | 439 | ), |
440 | | - enum=(self.enum.with_context(collisions=collisions) if self.enum else None), |
441 | | - meta=self.meta.with_context( |
442 | | - collisions=collisions, |
443 | | - ), |
| 440 | + enum=self.enum.with_context(collisions=collisions) if self.enum else None, |
| 441 | + meta=self.meta.with_context(collisions=collisions), |
444 | 442 | ) |
445 | 443 |
|
446 | 444 | def add_to_address_allowlist( |
@@ -741,9 +739,7 @@ def path_regex_str(self) -> str: |
741 | 739 | return parsing_regex_str |
742 | 740 |
|
743 | 741 | def get_field( |
744 | | - self, |
745 | | - *field_path: str, |
746 | | - collisions: Optional[Set[str]] = None, |
| 742 | + self, *field_path: str, collisions: Optional[Set[str]] = None |
747 | 743 | ) -> Field: |
748 | 744 | """Return a field arbitrarily deep in this message's structure. |
749 | 745 |
|
@@ -836,8 +832,7 @@ def with_context( |
836 | 832 | fields=( |
837 | 833 | { |
838 | 834 | k: v.with_context( |
839 | | - collisions=collisions, |
840 | | - visited_messages=visited_messages, |
| 835 | + collisions=collisions, visited_messages=visited_messages |
841 | 836 | ) |
842 | 837 | for k, v in self.fields.items() |
843 | 838 | } |
@@ -946,11 +941,7 @@ def ident(self) -> metadata.Address: |
946 | 941 | return self.meta.address |
947 | 942 |
|
948 | 943 | @cached_proto_context |
949 | | - def with_context( |
950 | | - self, |
951 | | - *, |
952 | | - collisions: Set[str], |
953 | | - ) -> "EnumType": |
| 944 | + def with_context(self, *, collisions: Set[str]) -> "EnumType": |
954 | 945 | """Return a derivative of this enum with the provided context. |
955 | 946 |
|
956 | 947 | This method is used to address naming collisions. The returned |
@@ -1997,9 +1988,7 @@ def with_context( |
1997 | 1988 | collisions=collisions, |
1998 | 1989 | visited_messages=visited_messages, |
1999 | 1990 | ), |
2000 | | - meta=self.meta.with_context( |
2001 | | - collisions=collisions, |
2002 | | - ), |
| 1991 | + meta=self.meta.with_context(collisions=collisions), |
2003 | 1992 | ) |
2004 | 1993 |
|
2005 | 1994 | def add_to_address_allowlist( |
@@ -2399,9 +2388,7 @@ def with_context( |
2399 | 2388 | ) |
2400 | 2389 | for k, v in self.methods.items() |
2401 | 2390 | }, |
2402 | | - meta=self.meta.with_context( |
2403 | | - collisions=collisions, |
2404 | | - ), |
| 2391 | + meta=self.meta.with_context(collisions=collisions), |
2405 | 2392 | ) |
2406 | 2393 |
|
2407 | 2394 | def add_to_address_allowlist( |
|
0 commit comments