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)