Migrate remaining PIL features away (#628)

Author: njooma

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "viam-sdk"
-version = "0.21.0"
+version = "0.21.1"
 description = "Viam Robotics Python SDK"
 authors = [ "Naveed <[email protected]>" ]
 license = "Apache-2.0"

src/viam/media/utils/pil.py renamed to src/viam/media/utils/pil/__init__.py

@@ -2,9 +2,8 @@
 
 from PIL import Image
 
-from viam.media.video import CameraMimeType, ViamImage
-
-from ..viam_rgba_plugin import RGBA_FORMAT_LABEL
+from ....media.video import CameraMimeType, ViamImage
+from .viam_rgba_plugin import RGBA_FORMAT_LABEL
 
 # Formats that are supported by PIL
 LIBRARY_SUPPORTED_FORMATS = ["JPEG", "PNG", RGBA_FORMAT_LABEL]

src/viam/media/viam_rgba_plugin.py renamed to src/viam/media/utils/pil/viam_rgba_plugin.py

@@ -4,16 +4,7 @@
 from PIL.ImageFile import ImageFile, PyDecoder, PyEncoder, _safe_read  # type: ignore -- (njooma) this exists, manually checked
 from PIL.ImageFile import _save as image_save  # type: ignore -- (njooma) this exists, manually checked
 
-# Viam uses a special header prepended to raw RGBA data. The header is composed of a
-# 4-byte magic number followed by a 4-byte line of the width as a uint32 number
-# and another for the height. Credit to Ben Zotto for inventing this formulation
-# https://bzotto.medium.com/introducing-the-rgba-bitmap-file-format-4a8a94329e2c
-
-RGBA_MAGIC_NUMBER = bytes("RGBA", "utf-8")
-
-RGBA_FORMAT_LABEL = "VIAM_RGBA"
-
-RGBA_HEADER_LENGTH = 12
+from ...viam_rgba import RGBA_FORMAT_LABEL, RGBA_HEADER_LENGTH, RGBA_MAGIC_NUMBER
 
 
 def _accept(prefix: str):

src/viam/media/viam_rgba.py

@@ -0,0 +1,10 @@
+# Viam uses a special header prepended to raw RGBA data. The header is composed of a
+# 4-byte magic number followed by a 4-byte line of the width as a uint32 number
+# and another for the height. Credit to Ben Zotto for inventing this formulation
+# https://bzotto.medium.com/introducing-the-rgba-bitmap-file-format-4a8a94329e2c
+
+RGBA_MAGIC_NUMBER = bytes("RGBA", "utf-8")
+
+RGBA_FORMAT_LABEL = "VIAM_RGBA"
+
+RGBA_HEADER_LENGTH = 12

src/viam/media/video.py

@@ -7,10 +7,7 @@
 from viam.errors import NotSupportedError
 from viam.proto.component.camera import Format
 
-from .viam_rgba_plugin import RGBA_FORMAT_LABEL, RGBA_HEADER_LENGTH, RGBA_MAGIC_NUMBER
-
-# Formats that are supported by PIL
-LIBRARY_SUPPORTED_FORMATS = ["JPEG", "PNG", RGBA_FORMAT_LABEL]
+from .viam_rgba import RGBA_HEADER_LENGTH, RGBA_MAGIC_NUMBER
 
 
 class CameraMimeType(str, Enum):