能否在类中导入函数提升可读性？迭代类重复代码复用咨询

Great question! This is exactly the kind of problem that Python's mixin classes were made to solve—you absolutely can write those reusable iterable methods once and reuse them across all your classes, cleaning up redundant code and letting each class focus on its unique functionality.

The Mixin Approach #

A mixin is a lightweight class that exists solely to provide reusable methods to other classes via inheritance. It’s not meant to be instantiated on its own; instead, you inherit from it alongside any other base classes your custom collection needs.

Here’s how to implement it:

Step 1: Create the Iterable Mixin #

First, bundle all your repetitive iterable methods into a single mixin class:

class IterableMixin:
    def __init__(self):
        self._items = []  # Use a private attribute to avoid naming conflicts
    
    def append(self, item):
        self._items.append(item)
    
    def __len__(self):
        return len(self._items)
    
    def __getitem__(self, index):
        return self._items[index]
    
    def __setitem__(self, index, value):
        self._items[index] = value
    
    def __iter__(self):
        self._current_idx = 0
        return self
    
    def __next__(self):
        if self._current_idx < len(self._items):
            item = self._items[self._current_idx]
            self._current_idx += 1
            return item
        raise StopIteration

Step 2: Inherit the Mixin in Your Custom Classes #

Now, any class that needs these iterable features can simply inherit from IterableMixin—no need to redefine those methods! You can focus entirely on the class’s unique functionality:

class UserCollection(IterableMixin):
    def __init__(self):
        super().__init__()  # Call the mixin's __init__ to initialize _items
        # Add class-specific initialization here
    
    def filter_admins(self):
        # Unique method for this class
        return [user for user in self if user.is_admin]


class ProductInventory(IterableMixin):
    def calculate_total_value(self):
        # Unique method for this class
        return sum(product.price for product in self)

Key Notes for Using Mixins #

• Prioritize Mixins for Code Reuse: Mixins are ideal for this scenario because they let you inject methods directly into your class’s namespace, keeping usage clean and intuitive.
• Override When Needed: If a specific class needs a modified version of a mixin method (e.g., a custom append that validates inputs), just redefine the method in the subclass—Python will prioritize the subclass’s implementation:

  class NumericCollection(IterableMixin):
      def append(self, item):
          if not isinstance(item, (int, float)):
              raise ValueError("Only numbers allowed!")
          super().append(item)  # Call the mixin's original append
  

• Multiple Inheritance Works: If your class already inherits from another base class, just add IterableMixin to the inheritance list (order matters if there are method conflicts—Python uses the Method Resolution Order, or MRO, to pick which method to use):

  class SpecializedCollection(SomeBaseClass, IterableMixin):
      # Your unique code here
  

Alternative: Composition (If Mixins Don’t Fit) #

If you prefer to avoid inheritance, you can use composition—wrap the reusable logic in a helper class and delegate methods to it:

class IterableHelper:
    def __init__(self):
        self._items = []
    
    # Same methods as the mixin: append, __len__, __getitem__, etc.

class BookCollection:
    def __init__(self):
        self._iter_helper = IterableHelper()
    
    # Delegate methods to the helper
    def append(self, book):
        self._iter_helper.append(book)
    
    def __len__(self):
        return self._iter_helper.__len__()
    
    # ... delegate other iterable methods similarly
    
    def find_by_author(self, author):
        # Unique functionality
        return [book for book in self._iter_helper if book.author == author]

This is more verbose but avoids inheritance-related complexities if you need strict separation of concerns.

内容的提问来源于stack exchange，提问作者M.G.Poirot