SerialConsole: compress screentshot to save space

The BMP file spends a lot of memory, convert it to PNG save about 600KB
per file.

lisa/sut_orchestrator/azure/features.py

@@ -3,12 +3,14 @@
 
 
 from dataclasses import dataclass
+from os import unlink
 from pathlib import Path
 from typing import TYPE_CHECKING, Any, Dict, List, Optional, Tuple, Type
 
 import requests
 from assertpy import assert_that
 from dataclasses_json import dataclass_json
+from PIL import Image
 
 from lisa import features, schema, search_space
 from lisa.features.gpu import ComputeSDK
@@ -76,12 +78,16 @@ class SerialConsole(AzureFeatureMixin, features.SerialConsole):
             )
         )
         if saved_path:
-            screenshot_name = saved_path.joinpath("serial_console.bmp")
+            screenshot_raw_name = saved_path.joinpath("serial_console.bmp")
+            screenshot_name = saved_path.joinpath("serial_console.png")
             screenshot_response = requests.get(
                 diagnostic_data.console_screenshot_blob_uri
             )
-            with open(screenshot_name, mode="wb") as f:
+            with open(screenshot_raw_name, mode="wb") as f:
                 f.write(screenshot_response.content)
+            with Image.open(screenshot_raw_name) as image:
+                image.save(screenshot_name, "PNG", optimize=True)
+            unlink(screenshot_raw_name)
 
         log_response = requests.get(diagnostic_data.serial_console_log_blob_uri)
 

poetry.lock

@@ -819,7 +819,7 @@ flake8-polyfill = ">=1.0.2,<2"
 name = "pillow"
 version = "8.3.1"
 description = "Python Imaging Library (Fork)"
-category = "dev"
+category = "main"
 optional = false
 python-versions = ">=3.6"
 
@@ -1553,6 +1553,14 @@ python-versions = "*"
 [package.dependencies]
 types-cryptography = "*"
 
+[[package]]
+name = "types-pillow"
+version = "8.3.3"
+description = "Typing stubs for Pillow"
+category = "dev"
+optional = false
+python-versions = "*"
+
 [[package]]
 name = "types-python-dateutil"
 version = "0.1.4"
@@ -1656,7 +1664,7 @@ watchmedo = ["PyYAML (>=3.10)", "argh (>=0.24.1)"]
 [metadata]
 lock-version = "1.1"
 python-versions = "^3.8"
-content-hash = "06c9394bb194c3c284e52f0534cad3880706c4a7c9e5d294567a1655b23539cc"
+content-hash = "2d1703132ca20756367b97808b20f7ce48d54f2f7bab52f07afe4c96e38ed136"
 
 [metadata.files]
 alabaster = [
@@ -1751,6 +1759,11 @@ cffi = [
     {file = "cffi-1.14.6-cp27-cp27m-win_amd64.whl", hash = "sha256:7bcac9a2b4fdbed2c16fa5681356d7121ecabf041f18d97ed5b8e0dd38a80224"},
     {file = "cffi-1.14.6-cp27-cp27mu-manylinux1_i686.whl", hash = "sha256:ed38b924ce794e505647f7c331b22a693bee1538fdf46b0222c4717b42f744e7"},
     {file = "cffi-1.14.6-cp27-cp27mu-manylinux1_x86_64.whl", hash = "sha256:e22dcb48709fc51a7b58a927391b23ab37eb3737a98ac4338e2448bef8559b33"},
+    {file = "cffi-1.14.6-cp35-cp35m-macosx_10_9_x86_64.whl", hash = "sha256:aedb15f0a5a5949ecb129a82b72b19df97bbbca024081ed2ef88bd5c0a610534"},
+    {file = "cffi-1.14.6-cp35-cp35m-manylinux1_i686.whl", hash = "sha256:48916e459c54c4a70e52745639f1db524542140433599e13911b2f329834276a"},
+    {file = "cffi-1.14.6-cp35-cp35m-manylinux1_x86_64.whl", hash = "sha256:f627688813d0a4140153ff532537fbe4afea5a3dffce1f9deb7f91f848a832b5"},
+    {file = "cffi-1.14.6-cp35-cp35m-win32.whl", hash = "sha256:f0010c6f9d1a4011e429109fda55a225921e3206e7f62a0c22a35344bfd13cca"},
+    {file = "cffi-1.14.6-cp35-cp35m-win_amd64.whl", hash = "sha256:57e555a9feb4a8460415f1aac331a2dc833b1115284f7ded7278b54afc5bd218"},
     {file = "cffi-1.14.6-cp36-cp36m-macosx_10_9_x86_64.whl", hash = "sha256:e8c6a99be100371dbb046880e7a282152aa5d6127ae01783e37662ef73850d8f"},
     {file = "cffi-1.14.6-cp36-cp36m-manylinux1_i686.whl", hash = "sha256:19ca0dbdeda3b2615421d54bef8985f72af6e0c47082a8d26122adac81a95872"},
     {file = "cffi-1.14.6-cp36-cp36m-manylinux1_x86_64.whl", hash = "sha256:d950695ae4381ecd856bcaf2b1e866720e4ab9a1498cba61c602e56630ca7195"},
@@ -2618,6 +2631,10 @@ types-paramiko = [
     {file = "types-paramiko-0.1.7.tar.gz", hash = "sha256:51ce59aea222c47d0590e2b8d8cefbc5aeb469d8bf6368e4f06ad3c61c932c72"},
     {file = "types_paramiko-0.1.7-py2.py3-none-any.whl", hash = "sha256:9fb9f3a363c99ae735148a7ce3c28adbe8f87797ec3c8cce3103899acbdebfa8"},
 ]
+types-pillow = [
+    {file = "types-Pillow-8.3.3.tar.gz", hash = "sha256:056b5385167284555714ab499ac2a4f81f6bbb9babf9e5b24c886e6c54ddc6ef"},
+    {file = "types_Pillow-8.3.3-py3-none-any.whl", hash = "sha256:54de6632d031f1cdde670182c5aedbddf5427418acf8325235d6767f9214355b"},
+]
 types-python-dateutil = [
     {file = "types-python-dateutil-0.1.4.tar.gz", hash = "sha256:e6486ca27b6dde73e0ec079a9e1b03e208766e6bc7f1e08964a7e9104a5c7d7a"},
     {file = "types_python_dateutil-0.1.4-py2.py3-none-any.whl", hash = "sha256:39bfe0bde61fc673b8fa28167bd78622d976210f791971b9f3e10877cbf119a4"},

pyproject.toml

@@ -29,6 +29,7 @@ spurplus = "^2.3.4"
 semver = "^2.13.0"
 pep8-naming = "^0.12.0"
 types-toml = "^0.1.5"
+Pillow = "^8.3.1"
 
 [tool.poetry.dev-dependencies]
 black = "^21.7b0"
@@ -55,6 +56,7 @@ types-python-dateutil = "^0.1.4"
 types-PyYAML = "^5.4.3"
 rstcheck = "^3.3.1"
 snooty-lextudio = "^1.9.2-alpha.0"
+types-Pillow = "^8.3.3"
 
 [tool.black]
 line-length = 88