Create sample code for UserList vs list

Author: lpozo

python-inherit-list-userlist/README.md

@@ -0,0 +1,4 @@
+# Custom Python Lists: Inheriting From list vs UserList
+
+This folder provides the code examples for the article [Custom Python Lists: Inheriting From list vs UserList
+](https://realpython.com/inherit-python-list/).

python-inherit-list-userlist/custom_list1.py

@@ -0,0 +1,13 @@
+class CustomList(list):
+    def join(self, sep=" "):
+        return sep.join(str(item) for item in self)
+
+    def map(self, action):
+        return type(self)(action(item) for item in self)
+
+    def filter(self, predicate):
+        return type(self)(item for item in self if predicate(item))
+
+    def for_each(self, func):
+        for item in self:
+            yield func(item)

python-inherit-list-userlist/custom_list2.py

@@ -0,0 +1,16 @@
+from collections import UserList
+
+
+class CustomList(UserList):
+    def join(self, sep=" "):
+        return sep.join(self.data)
+
+    def map(self, action):
+        return type(self)(action(item) for item in self.data)
+
+    def filter(self, predicate):
+        return type(self)(item for item in self if predicate(item))
+
+    def for_each(self, func):
+        for item in self.data:
+            yield func(item)

python-inherit-list-userlist/number_list1.py

@@ -0,0 +1,23 @@
+class NumberList(list):
+    def __init__(self, iterable):
+        super().__init__(self._validate_number(item) for item in iterable)
+
+    def __setitem__(self, index, item):
+        super().__setitem__(index, self._validate_number(item))
+
+    def insert(self, index, item):
+        super().insert(index, self._validate_number(item))
+
+    def append(self, item):
+        super().append(self._validate_number(item))
+
+    def extend(self, other):
+        if isinstance(other, type(self)):
+            super().extend(other)
+        else:
+            super().extend(self._validate_number(item) for item in other)
+
+    def _validate_number(self, value):
+        if isinstance(value, (int, float, complex)):
+            return value
+        raise TypeError(f"numeric value expected, got {type(value).__name__}")

python-inherit-list-userlist/number_list2.py

@@ -0,0 +1,26 @@
+from collections import UserList
+
+
+class NumberList(UserList):
+    def __init__(self, iterable):
+        self.data = [self._validate_number(item) for item in iterable]
+
+    def __setitem__(self, index, item):
+        self.data[index] = self._validate_number(item)
+
+    def insert(self, index, item):
+        self.data.insert(index, self._validate_number(item))
+
+    def append(self, item):
+        self.data.append(self._validate_number(item))
+
+    def extend(self, other):
+        if isinstance(other, type(self)):
+            self.data.extend(other)
+        else:
+            self.data.extend(self._validate_number(item) for item in other)
+
+    def _validate_number(self, value):
+        if isinstance(value, (int, float, complex)):
+            return value
+        raise TypeError(f"numeric value expected, got {type(value).__name__}")

python-inherit-list-userlist/performance.py

@@ -0,0 +1,94 @@
+import timeit
+from collections import UserList
+
+
+class StringList_list(list):
+    def __init__(self, iterable):
+        super().__init__(str(item) for item in iterable)
+
+    def __setitem__(self, index, item):
+        super().__setitem__(index, str(item))
+
+    def insert(self, index, item):
+        super().insert(index, str(item))
+
+    def append(self, item):
+        super().append(str(item))
+
+    def extend(self, other):
+        if isinstance(other, type(self)):
+            super().extend(other)
+        else:
+            super().extend(str(item) for item in other)
+
+
+class StringList_UserList(UserList):
+    def __init__(self, iterable):
+        self.data = [str(item) for item in iterable]
+
+    def __setitem__(self, index, item):
+        self.data[index] = str(item)
+
+    def insert(self, index, item):
+        self.data.insert(index, str(item))
+
+    def append(self, item):
+        self.data.append(str(item))
+
+    def extend(self, other):
+        if isinstance(other, type(self)):
+            self.data.extend(other)
+        else:
+            self.data.extend(str(item) for item in other)
+
+
+init_data = range(10000)
+
+list_initialization = min(
+    timeit.repeat(
+        stmt="StringList_list(init_data)",
+        number=1000,
+        repeat=5,
+        globals=globals(),
+    )
+)
+
+user_list_initialization = min(
+    timeit.repeat(
+        stmt="StringList_UserList(init_data)",
+        number=1000,
+        repeat=5,
+        globals=globals(),
+    )
+)
+
+print(
+    f"list is {list_initialization / user_list_initialization:.3f}",
+    "times slower than UserList",
+)
+
+
+extended_list = StringList_list(init_data)
+list_extend = min(
+    timeit.repeat(
+        stmt="extended_list.extend(init_data)",
+        number=5,
+        repeat=2,
+        globals=globals(),
+    )
+)
+
+extended_user_list = StringList_UserList(init_data)
+user_list_extend = min(
+    timeit.repeat(
+        stmt="extended_user_list.extend(init_data)",
+        number=5,
+        repeat=2,
+        globals=globals(),
+    )
+)
+
+print(
+    f"list is {list_extend / user_list_extend:.3f}",
+    "times slower than UserList",
+)

python-inherit-list-userlist/string_list1.py

@@ -0,0 +1,21 @@
+from collections import UserList
+
+
+class StringList(UserList):
+    def __init__(self, iterable):
+        super().__init__(str(item) for item in iterable)
+
+    def __setitem__(self, index, item):
+        super().__setitem__(index, str(item))
+
+    def insert(self, index, item):
+        super().insert(index, str(item))
+
+    def append(self, item):
+        super().append(str(item))
+
+    def extend(self, other):
+        if isinstance(other, type(self)):
+            super().extend(other)
+        else:
+            super().extend(str(item) for item in other)

python-inherit-list-userlist/string_list2.py

@@ -0,0 +1,21 @@
+from collections import UserList
+
+
+class StringList(UserList):
+    def __init__(self, iterable):
+        self.data = [str(item) for item in iterable]
+
+    def __setitem__(self, index, item):
+        self.data[index] = str(item)
+
+    def insert(self, index, item):
+        self.data.insert(index, str(item))
+
+    def append(self, item):
+        self.data.append(str(item))
+
+    def extend(self, other):
+        if isinstance(other, type(self)):
+            self.data.extend(other)
+        else:
+            self.data.extend(str(item) for item in other)