Skip to content

Commit 5566014

Browse files
committed
fix mypy
1 parent c1276b9 commit 5566014

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

src/opengeodeweb_viewer/rpc/generic/generic_protocols.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -29,9 +29,10 @@ def register(self, params):
2929
data_id = str(params["id"])
3030
specific_params = {"id": data_id}
3131
data = self.get_data(data_id)
32-
if data.viewer_object == "mesh":
32+
viewer_object = str(data["viewer_object"])
33+
if viewer_object == "mesh":
3334
self.mesh_protocols.registerMesh(specific_params)
34-
elif data.viewer_object == "model":
35+
elif viewer_object == "model":
3536
self.model_protocols.registerModel(specific_params)
3637

3738
@exportRpc(generic_prefix + generic_schemas_dict["deregister"]["rpc"])
@@ -42,7 +43,8 @@ def deregister(self, params):
4243
data_id = str(params["id"])
4344
specific_params = {"id": data_id}
4445
data = self.get_data(data_id)
45-
if data.viewer_object == "mesh":
46+
viewer_object = str(data["viewer_object"])
47+
if viewer_object == "mesh":
4648
self.mesh_protocols.deregisterMesh(specific_params)
47-
elif data.viewer_object == "model":
49+
elif viewer_object == "model":
4850
self.model_protocols.deregisterModel(specific_params)

0 commit comments

Comments
 (0)