refactoring

2020-03-07 20:51:54 +04:00
parent d6a685887f
commit 3b6ad4abf9
4 changed files with 65 additions and 52 deletions
@@ -46,9 +46,11 @@ class IEPolys:

    def add(self, type):
        self.list = self.list[0:self.n]
-        self.list.append ( IEPolysPoints(self, type) )
+        l = IEPolysPoints(self, type)
+        self.list.append ( l )
        self.n_max = self.n = self.n + 1
        self.dirty = True
+        return l

    def n_dec(self):
        self.n = max(0, self.n-1)