From 28c4c27d2b81691e7e5403ffa064191301043944 Mon Sep 17 00:00:00 2001 From: Taka Date: Mon, 5 Oct 2020 09:08:42 +0000 Subject: [PATCH] Saved the generated ASA screenshot after shrinking it. --- vasl_templates/webapp/scenarios.py | 14 ++++++++++---- 1 file changed, 10 insertions(+), 4 deletions(-) diff --git a/vasl_templates/webapp/scenarios.py b/vasl_templates/webapp/scenarios.py index 10b6943..efb23e0 100644 --- a/vasl_templates/webapp/scenarios.py +++ b/vasl_templates/webapp/scenarios.py @@ -465,10 +465,6 @@ def prepare_asa_upload(): #pylint: disable=too-many-locals ) # read the screenshot image - screenshot_file.save_copy( - app.config.get( "PREPARE_ASA_UPLOAD_SCREENSHOT" ), - logger, "generated screenshot" - ) if os.path.getsize( screenshot_file.name ) == 0: # NOTE: The VASSAL shim sometimes crashes while trying to generate a screenshot :-( screenshot_data = None @@ -498,6 +494,16 @@ def prepare_asa_upload(): #pylint: disable=too-many-locals # get the final image data screenshot_data = save_image( img ) + # save a copy of the screenshot image + fname = app.config.get( "PREPARE_ASA_UPLOAD_SCREENSHOT" ) + if fname: + logger.debug( "Saving a copy of the generated screenshot: %s", fname ) + with open( fname, "wb" ) as fp: + fp.write( screenshot_data ) + else: + if os.path.isfile( fname ): + os.unlink( fname ) + except Exception as ex: #pylint: disable=broad-except return VassalShim.translate_vassal_shim_exception( ex, logger )