Pythonic Dependency Injection

gui commited 3 years ago · ๐Ÿ Python dependency-injection solid

EDIT 2022/April Today Python supports Protocols which is a better alternative to this problem.

Introduction

The first thing that I thought that was missing when I came from a C# background to Python was:

What should I use for Dependency Injection? How can I create interfaces?

And looks like that's a pretty common question from anyone coming from a typed language background, based on these stack overflow questions:

Please note how all these questions always mention: "Well, in Java/C#/another language...".

Let me interrupt it right there because, well, you're not in Java, C#, or something else, so how should you face this problem?

๐Ÿ“– Glossary Cheat Sheet

Before we dive into it, let's review some acronyms and what they represent:

๐Ÿค” What's the Dependency Inversion Principle and why should I care?

Let me make it simple, inversion of control states that your code should respect two rules:

DIP concept

In other words, you can achieve that in C# with something like:

    public interface IDependency
    {
            void work();
    }

    public class Implementation : IDependency
    {
            public void work()
            {
                    Console.WriteLine("Implementation of Work!");
            }
    }

    public class Module
    {
            private IDependency dependency;

            public Module(dependency: IDependency)
            {
                    this.dependency = dependency;
            }

            public doWork()
            {
                    this.dependency.work();
            }
    }

Ok, note how Module knows nothing about Implementation, and how Implementation knows nothing about Module - Instead, both of them know only IDependency which can be seen as a contract between them that says: "Hey, you should implement method work with no args and return void".

Cool, got it, why should I care? Well, it brings us a couple of benefits:

This makes your code insanely scalable, you should write code like that always.

(Honestly, I can't imagine a high quality C#/Java code that does not respect this rule.)

โ›“๏ธ Python does not have interfaces

That's damn tricky, how am I supposed to depend on abstractions if Python doesn't support interfaces?

Well, there's the ABC module that would allow you to write an abstract class disguised as an interface. I don't believe Python needs interfaces.

That's somewhat shocking, after researching a lot on StackOverflow and other posts, I came up to a conclusion:

Interface is a solution for languages that do not support multiple inheritance.

And guess what? Python supports it (I won't talk about the diamond problem caused by such decision).

Languages that allow only single inheritance, where a class can only derive from one base class, do not have the diamond problem. The reason for this is that such languages have at most one implementation of any method at any level in the inheritance chain regardless of the repetition or placement of methods. Typically these languages allow classes to implement multiple protocols, called interfaces in Java. These protocols define methods but do not provide concrete implementations. This strategy has been used by ActionScript, C#, D, Java, Nemerle, Object Pascal, Objective-C, Smalltalk, Swift and PHP.[13] All these languages allow classes to implement multiple protocols. โ€“ From Wikipedia

๐Ÿ‘ป What about abstract classes?

An Abstract Class is a way of sharing behaviors and implementations between its subclasses and also enforcing some contract between subclasses. You should use it if required.

My advice is not against abstract classes but "pure" disguised abstract classes as interfaces.

Ok, but how can I enforce a contract, then? How can I guarantee that my implementation has all the required methods and details?

Well, that brings us to the next topic, which is:

๐Ÿฆ† Python is dynamically typed

Man, have you realized that you need to specify types in some languages, but not in python? (type hinting does not enforce it btw).

Look at the following Python function:

def sum_join(v1, v2):
    return v1 + v2

This function does not care about what passes, it will return the "sum" of 2 arguments.

This means that you can use it like:

sum_join(1, 1)     # produces 2
sum_join("a", "b") # produces "ab"

Such thing is called duck typing, a concept that says: โ€œif it quacks like a duck then it must be a duckโ€.

It means that your code assumes that having the methods/functions that you need means your implementation is fine.

That's very powerful, so I'll repeat the question:

How can I enforce a contract? How can I rest assured that my code has implemented everything needed, and for God's sake it's taking args and returning correctly?

Well, you can't. At least, not with regular classes.

Instead, I would ask you (regardless of how you enforce things) how would you prove to me that your code implementation is correct and works for real?

๐Ÿงช Unit Testing is the key

In the same way you SHOULD test that a function returns the expected result, you should test that your implementation works fine in your code.

You don't need an interpreter to force you to do stuff. You're a grown-up and you can ensure you did it yourself.

Let me try to prove my point because talk is cheap.

Let's consider we have a CronJob that runs every hour and we want to be notified somehow whenever it gets executed.

class CronJob:
    def __init__(self, notificator):
        self.notificator = notificator

    def execute(self):
        print("Executing cron job")
        notification_emitted = self.notificator.send("Job has been executed")
        if not notification_emitted:
            raise Exception("Notification failed")

Note that we're receiving a "notificator" and it should contain a send method - and that's it.

On C# you would implement it as:

public interface INotificator
{
    bool send(String message);
}

public class WhatsAppNotification : INotificator
{
        public bool send(String message)
        {
                Console.WriteLine("Sending notification through Whatsapp");
                return true;
        }
}

Now in python, let's imagine we have 3 implementations with 1 not respecting our send method.

class WhatsAppNotification:
    def send(self, msg):
        print("Sending notification through Whatsapp")
        return True


class EmailNotification:
    def send(self, msg):
        print("Sending notification through Email")
        return True


class TelegramNotification:
    def notificate(self, msg):
        print("Not respecting contract")

How can we ensure we're not sending this broken TelegramNotification into production and ruining the whole thing?

We can have these unit tests to ensure everything works:

from unittest.mock import patch
import pytest
import notifications
from jobs import CronJob


class TestValidContract:
    """Test job dependency with different implementations
    """
    def test_job_notificates_through_whatsapp(self):
        job = CronJob(notifications.WhatsAppNotification())
        job.execute()

    def test_job_notificates_through_email(self):
        job = CronJob(notifications.EmailNotification())
        job.execute()

    def test_job_throws_exception_when_notification_fails(self):
        """Mocks notificator to return False
        """
        notificator = notifications.WhatsAppNotification()
        with patch.object(notificator, "send", return_value=False):
            with pytest.raises(Exception):
                job = CronJob(notificator)
                job.execute()


class TestBrokenContract:
    """Suite of tests that prove Telegram is not respecting
    its contract.
    """

    def test_telegram_throws_exception_when_notification_fails(self):
        """Mocking notificator does not fix a broken
        contract
        """
        notificator = notifications.TelegramNotification()
        with patch.object(notificator, "send", return_value=False):
            with pytest.raises(Exception):
                job = CronJob(notificator)
                job.execute()

    def test_job_notificates_through_telegram(self):
        """This test will fail because
        Telegram implementation does not respect contract
        """
        job = CronJob(notifications.TelegramNotification())
        job.execute()

I decided to split the working suite from the failing one to make it easier to spot.

Note what's really interesting

But hey, don't trust me, please go straight to the repository, clone, execute, and see it by yourself on GitHub

I'm adding to the source code a scenario where I believe it would be fine to implement abstract classes as well.

๐Ÿ Can we implement Dependency Injection in Python?

Yes! but that may depend on how you see things.

The inversion of Control Principle states that:

and that's how I honestly see it now:

DIP languages

A python dynamic type that can be anything is the abstraction.

So, based on our examples, that's how we can see it:

DIP example

notificator (the argument inside CronJob), is an abstraction and WhatsAppNotification is the actual implementation.

But please note, CronJob receives an implementation through the abstraction, that's why we're respecting the principle.

Examples

Abstract classes

I have absolutely nothing against abstract classes, I use them quite often by the way. I'm just opposed to writing more lines of code that don't add value. If you're writing unit tests (I really hope you are) why enforce the same thing twice?

If you have an actual behavior that you want to share, then it starts making sense to me. You need to ensure a base class has something to share, without deciding details of the implementation.

(See this example for a valid reason for using abstract classes: GitHub).

Dependency Injection on classes

As the example showed above, it does respect the principle:

class CronJob:
    def __init__(self, notificator):
        self.notificator = notificator

By allowing our CronJob to receive anything that's perfectly ok. You might also provide a default, so you can do it like:

class CronJob:
    def __init__(self, notificator=None):
        self.notificator = notificator or WhatsAppNotification()

If you allow your code to take ANY implementation, then IMHO it still respects this principle.

Dependency Injection on modules

Different from other languages python has "modules" that can contain several functions.

In such scenario there's no constructor to inject the implementation.

So I often end up doing:

notificator = WhatsAppNotification()

def execute_job():
    notificator.send("Job executed")

See? notificator is now a module dependency, and it can be anything thanks to Duck Typing. We need to implement unit tests to ensure that our notificator respects the send contract.

KISP: Keep It Simple, Pythonista - no IoC Containers

Again, if you come from other languages, you probably are familiar with a few IoC Containers.

I can't see a good reason to implement them in Python. Your code gets verbose, new developers don't really understand what's happening behind your code (honestly neither do I lol).

Why to add more lines to your code if you can achieve:

without an IoC framework?

โญ Conclusion

Let's wrap it up:

๐Ÿ”– References

I relied on a few other article to write this one:

  • LinkedIn
  • Tumblr
  • Reddit
  • Google+
  • Pinterest
  • Pocket