Skip to content

Commit bb1d541

Browse files
committed
feat: expose Registry on Plugin
1 parent 63d657b commit bb1d541

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

protogen/__init__.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1464,15 +1464,19 @@ class Plugin:
14641464
files_to_generate : List[File]
14651465
Set of files to code generation is request for. These are the files
14661466
explictly passed to protoc as command line arguments.
1467+
registry : Registry
1468+
The registry that was used in the resolution process for this plugin.
14671469
"""
14681470

14691471
def __init__(
14701472
self,
14711473
parameter: Dict[str, str],
14721474
files_to_generate: List[File],
1475+
registry: Registry,
14731476
):
14741477
self.parameter = parameter
14751478
self.files_to_generate = files_to_generate
1479+
self.registry = registry
14761480

14771481
self._error: Optional[str] = None
14781482
self._generated_files: List[GeneratedFile] = []
@@ -1668,7 +1672,7 @@ def run(self, f: Callable[[Plugin], None]):
16681672
files_to_generate.append(file)
16691673

16701674
# Create plugin and run the provided code generation function.
1671-
plugin = Plugin(parameter, files_to_generate)
1675+
plugin = Plugin(parameter, files_to_generate, registry)
16721676
f(plugin)
16731677

16741678
# Write response.

0 commit comments

Comments
 (0)