From cf004ec5de3092ba294395cbe1c8d1d74b0db6bd Mon Sep 17 00:00:00 2001 From: Pascal Lais Date: Sun, 22 Mar 2020 18:40:54 +0100 Subject: [PATCH] Add __getitem__ and __setitem__ for linesegment --- linesegment.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/linesegment.py b/linesegment.py index 8968f52..81ab989 100644 --- a/linesegment.py +++ b/linesegment.py @@ -56,6 +56,20 @@ class LineSegment: def get_max_y(self): """Get the highest y coordinate of the line segment""" return max(self.__startpoint.get_y(), self.__endpoint.get_y()) + def __getitem__(self, index): + """Get an Endpoint by index""" + if index == 0: + return self.__startpoint + elif index == 1: + return self.__endpoint + else: + return None + def __setitem__(self, index, value): + """Set an Endpoint by index""" + if index == 0: + self.set_startpoint(value) + elif index == 1: + self.set_endpoint(value) def __str__(self): """Return the line segment values as string.""" return str(self.__startpoint) + "-->" + str(self.__endpoint)