From 22dc6df1c791c912f3bac3654730f661fa0137ce Mon Sep 17 00:00:00 2001 From: Thomas Wouters Date: Thu, 24 Aug 2000 20:14:10 +0000 Subject: Support for augmented assignment in the UserList, UserDict, UserString and rfc822 (Addresslist) modules. Also a preliminary testcase for augmented assignment, which should actually be merged with the test_class testcase I added last week. --- Lib/UserList.py | 11 +++++++++++ 1 file changed, 11 insertions(+) (limited to 'Lib/UserList.py') diff --git a/Lib/UserList.py b/Lib/UserList.py index d4a8d2f6e4..114606098c 100644 --- a/Lib/UserList.py +++ b/Lib/UserList.py @@ -51,9 +51,20 @@ class UserList: return self.__class__(other + self.data) else: return self.__class__(list(other) + self.data) + def __iadd__(self, other): + if isinstance(other, UserList): + self.data += other.data + elif isinstance(other, type(self.data)): + self.data += other + else: + self.data += list(other) + return self def __mul__(self, n): return self.__class__(self.data*n) __rmul__ = __mul__ + def __imul__(self, n): + self.data *= n + return self def append(self, item): self.data.append(item) def insert(self, i, item): self.data.insert(i, item) def pop(self, i=-1): return self.data.pop(i) -- cgit v1.2.1