Add getCamshot (#20)

cgarwood · Jun 11, 2024 · ea197b7 · ea197b7
1 parent ec08261
commit ea197b7
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/fullykiosk/__init__.py b/fullykiosk/__init__.py
@@ -127,6 +127,9 @@ async def clearWebstorage(self):
     async def clearCookies(self):
         await self.sendCommand("clearCookies")
 
+    async def getCamshot(self):
+        return await self._rh.get(cmd="getCamshot", password=self._password)
+
 
 class _RequestsHandler:
     """Internal class to create FullyKiosk requests"""
@@ -161,10 +164,10 @@ async def get(self, **kwargs):
                 )
                 raise FullyKioskError(response.status, await response.text())
 
-            try:
-                data = await response.json()
-            except aiohttp.client_exceptions.ContentTypeError:
-                data = await response.json(content_type="text/html")
+            content_type = response.headers['Content-Type']
+            if content_type.startswith("image/"):
+                return await response.content.read()
+            data = await response.json(content_type=content_type)
 
             _LOGGER.debug(json.dumps(data))
             return data