Dead simple depedency injection container (IoC container) in Python

di_container.py

class DIContainer:
    bindings = {}

    def set(self, abstract, factory):
        self.bindings[abstract] = factory

    def get(self, abstract):
        return self.bindings[abstract](self)


class Logger:
    def __init__(self):
        pass

    def log(self):
        print("Log message")


class DBStorage:
    def __init__(self, logger: Logger):
        self.logger = logger

    def store(self):
        print("Store data")


class ShippingService:
    def __init__(self, db_storage: DBStorage):
        self.db_storage = db_storage

    def ship(self):
        print("Ship product")


container = DIContainer()

container.set(ShippingService, lambda container: ShippingService(container.get(DBStorage)))
container.set(DBStorage, lambda container: DBStorage(container.get(Logger)))
container.set(Logger, lambda container: Logger())

# shopping_service = ShippingService(DBStorage(Logger()))  # Without IoC container
shipping_service = container.get(ShippingService)
shipping_service.ship()