Pass betterproto option using custom_opt instead of environment variable

2020-06-06 12:51:37 +02:00
parent d31f90be6b
commit f7769a19d1
2 changed files with 8 additions and 5 deletions
--- a/betterproto/plugin.py
+++ b/betterproto/plugin.py
@@ -120,8 +120,7 @@ def get_comment(proto_file, path: List[int], indent: int = 4) -> str:


 def generate_code(request, response):
-    plugin_options = os.environ.get("BETTERPROTO_OPTS")
-    plugin_options = plugin_options.split(" ") if plugin_options else []
+    plugin_options = request.parameter.split(",") if request.parameter else []

    env = jinja2.Environment(
        trim_blocks=True,