os environ changes

Author: shriram192

servicex_codegen/__init__.py

@@ -61,6 +61,8 @@ def create_app(test_config=None, provided_translator=None):
         if 'CODEGEN_CONFIG_FILE' in os.environ:
             app.config.from_envvar('CODEGEN_CONFIG_FILE')
 
+        app.config['TRANSFORMER_SCIENCE_IMAGE'] = os.environ.get('TRANSFORMER_SCIENCE_IMAGE')
+
     with app.app_context():
         translator = provided_translator
 

servicex_codegen/post_operation.py

@@ -32,10 +32,11 @@
 # modification, are permitted provided that the following conditions are met:
 #
 import os
+
 import zipfile
 from tempfile import TemporaryDirectory
 
-from flask import Response, request
+from flask import Response, request, current_app
 from flask_restful import Resource
 from requests_toolbelt import MultipartEncoder
 
@@ -92,7 +93,7 @@ def post(self):
                 zip_data = self.stream_generated_code(generated_code_result)
                 # code gen transformer returns the default transformer image mentioned in
                 # the config file
-                transformer_image = os.environ.get('TRANSFORMER_SCIENCE_IMAGE')
+                transformer_image = current_app.config.get("TRANSFORMER_SCIENCE_IMAGE")
 
                 # MultipartEncoder library takes multiple types of data fields and merge
                 # them into a multipart mime data type