Designing Flexible Code with Polymorphism in Python

Polymorphism is one of the cornerstones of object-oriented programming (OOP) that allows objects of different types to be treated as objects of a common type. By mastering polymorphism, you can design flexible, reusable, and maintainable Python code.

What is Polymorphism?

Polymorphism, derived from Greek meaning 'many forms', enables a single interface to represent different underlying forms (data types). In Python, polymorphism is implemented through method overriding and interfaces like abstract base classes.

Key Benefits of Polymorphism

Code Reusability: Write generic code that works across multiple classes.
Flexibility: Change behavior dynamically based on object type.
Maintainability: Easier to extend and debug code without modifying existing logic.

Implementing Polymorphism in Python

Let's look at an example where polymorphism is used to calculate the area of different shapes.

from abc import ABC, abstractmethod

class Shape(ABC):
    @abstractmethod
    def area(self):
        pass

class Circle(Shape):
    def __init__(self, radius):
        self.radius = radius

    def area(self):
        return 3.14 * self.radius ** 2

class Rectangle(Shape):
    def __init__(self, width, height):
        self.width = width
        self.height = height

    def area(self):
        return self.width * self.height

# Using polymorphism
def print_area(shape):
    print(f"The area is: {shape.area()}")

shapes = [Circle(5), Rectangle(4, 6)]
for shape in shapes:
    print_area(shape)

In this example, the print_area function demonstrates polymorphism by accepting any object that implements the area method.