From ad5a6a41531ed2011bddb4bff45ce2b3d12ef8cf Mon Sep 17 00:00:00 2001 From: Jaime van Kessel Date: Fri, 25 Jan 2019 14:09:41 +0100 Subject: [PATCH] Added missing typing --- cura/LayerDataDecorator.py | 18 ++++++++++++++---- 1 file changed, 14 insertions(+), 4 deletions(-) diff --git a/cura/LayerDataDecorator.py b/cura/LayerDataDecorator.py index c04479972a..568168a309 100644 --- a/cura/LayerDataDecorator.py +++ b/cura/LayerDataDecorator.py @@ -1,13 +1,23 @@ +from typing import Optional + from UM.Scene.SceneNodeDecorator import SceneNodeDecorator ## Simple decorator to indicate a scene node holds layer data. +from cura.LayerData import LayerData + + class LayerDataDecorator(SceneNodeDecorator): def __init__(self): super().__init__() - self._layer_data = None + self._layer_data = None # type: Optional[LayerData] - def getLayerData(self): + def getLayerData(self) -> Optional[LayerData]: return self._layer_data - def setLayerData(self, layer_data): - self._layer_data = layer_data \ No newline at end of file + def setLayerData(self, layer_data: LayerData) -> None: + self._layer_data = layer_data + + def __deepcopy__(self, memo) -> "LayerDataDecorator": + copied_decorator = LayerDataDecorator() + copied_decorator.setLayerData(self.getLayerData()) + return copied_decorator