Return text response if communicator cannot be fetched for ClearGameD…

…ataEndpoint.py
Map-A-Droid · Feb 21, 2024 · 5d42036 · 5d42036
1 parent 4ee0a2e
commit 5d42036
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/mapadroid/madmin/endpoints/routes/control/ClearGameDataEndpoint.py b/mapadroid/madmin/endpoints/routes/control/ClearGameDataEndpoint.py
@@ -6,6 +6,7 @@
 from mapadroid.madmin.endpoints.routes.control.AbstractControlEndpoint import \
     AbstractControlEndpoint
 from mapadroid.mapping_manager.MappingManager import DeviceMappingsEntry
+from mapadroid.websocket.AbstractCommunicator import AbstractCommunicator
 
 
 class ClearGameDataEndpoint(AbstractControlEndpoint):
@@ -29,6 +30,8 @@ async def get(self):
             pass
             # origin_logger.info('MADmin: ADB shell command successfully')
         else:
-            temp_comm = self._get_ws_server().get_origin_communicator(origin)
+            temp_comm: Optional[AbstractCommunicator] = self._get_ws_server().get_origin_communicator(origin)
+            if not temp_comm:
+                return web.Response(text="Failed fetching connection to device.")
             await temp_comm.reset_app_data("com.nianticlabs.pokemongo")
         raise web.HTTPFound(self._url_for("get_phonescreens"))