mirror of
				https://github.com/ets-labs/python-dependency-injector.git
				synced 2025-11-04 01:47:36 +03:00 
			
		
		
		
	adding scoped provider with examples
This commit is contained in:
		
							parent
							
								
									1631978790
								
							
						
					
					
						commit
						1cdf393f56
					
				
							
								
								
									
										67
									
								
								examples/scoped_provider.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										67
									
								
								examples/scoped_provider.py
									
									
									
									
									
										Normal file
									
								
							| 
						 | 
					@ -0,0 +1,67 @@
 | 
				
			||||||
 | 
					"""
 | 
				
			||||||
 | 
					Scoped provider examples.
 | 
				
			||||||
 | 
					"""
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					from objects import AbstractCatalog
 | 
				
			||||||
 | 
					from objects.providers import Singleton, Scoped
 | 
				
			||||||
 | 
					from objects.injections import InitArg, Attribute
 | 
				
			||||||
 | 
					import sqlite3
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					class ObjectA(object):
 | 
				
			||||||
 | 
					    def __init__(self, db):
 | 
				
			||||||
 | 
					        self.db = db
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					# Catalog of objects providers.
 | 
				
			||||||
 | 
					class Catalog(AbstractCatalog):
 | 
				
			||||||
 | 
					    """
 | 
				
			||||||
 | 
					    Objects catalog.
 | 
				
			||||||
 | 
					    """
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    database = Singleton(sqlite3.Connection,
 | 
				
			||||||
 | 
					                         InitArg('database', ':memory:'),
 | 
				
			||||||
 | 
					                         Attribute('row_factory', sqlite3.Row))
 | 
				
			||||||
 | 
					    """ :type: (objects.Provider) -> sqlite3.Connection """
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    object_a = Scoped(ObjectA,
 | 
				
			||||||
 | 
					                      InitArg('db', database))
 | 
				
			||||||
 | 
					    """ :type: (objects.Provider) -> ObjectA """
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					with Catalog.object_a as object_a_provider:
 | 
				
			||||||
 | 
					    object_a1 = object_a_provider()
 | 
				
			||||||
 | 
					    object_a2 = object_a_provider()
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    assert object_a1 is object_a2
 | 
				
			||||||
 | 
					    assert object_a1.db is object_a2.db
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					with Catalog.object_a as object_a_provider:
 | 
				
			||||||
 | 
					    object_a3 = object_a_provider()
 | 
				
			||||||
 | 
					    object_a4 = object_a_provider()
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    assert object_a3 is object_a4
 | 
				
			||||||
 | 
					    assert object_a3.db is object_a4.db
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    assert (object_a1 is not object_a3) and \
 | 
				
			||||||
 | 
					           (object_a1 is not object_a4)
 | 
				
			||||||
 | 
					    assert (object_a2 is not object_a3) and \
 | 
				
			||||||
 | 
					           (object_a2 is not object_a4)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					Catalog.object_a.in_scope()
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					object_a5 = Catalog.object_a()
 | 
				
			||||||
 | 
					object_a6 = Catalog.object_a()
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					assert object_a5 is object_a6
 | 
				
			||||||
 | 
					assert object_a5.db is object_a6.db
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					assert (object_a1 is not object_a3) and \
 | 
				
			||||||
 | 
					       (object_a1 is not object_a4) and \
 | 
				
			||||||
 | 
					       (object_a1 is not object_a5) and \
 | 
				
			||||||
 | 
					       (object_a1 is not object_a6)
 | 
				
			||||||
 | 
					assert (object_a2 is not object_a3) and \
 | 
				
			||||||
 | 
					       (object_a2 is not object_a4) and \
 | 
				
			||||||
 | 
					       (object_a2 is not object_a5) and \
 | 
				
			||||||
 | 
					       (object_a2 is not object_a6)
 | 
				
			||||||
| 
						 | 
					@ -108,6 +108,60 @@ class Singleton(NewInstance):
 | 
				
			||||||
            self.instance = super(Singleton, self).__call__(*args, **kwargs)
 | 
					            self.instance = super(Singleton, self).__call__(*args, **kwargs)
 | 
				
			||||||
        return self.instance
 | 
					        return self.instance
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    def _reset_instance(self):
 | 
				
			||||||
 | 
					        """
 | 
				
			||||||
 | 
					        Resets instance.
 | 
				
			||||||
 | 
					        """
 | 
				
			||||||
 | 
					        self.instance = None
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					class Scoped(Singleton):
 | 
				
			||||||
 | 
					    """
 | 
				
			||||||
 | 
					    Scoped provider will create instance once for every scope and return it on every call.
 | 
				
			||||||
 | 
					    """
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    def __init__(self, *args, **kwargs):
 | 
				
			||||||
 | 
					        """
 | 
				
			||||||
 | 
					        Initializer.
 | 
				
			||||||
 | 
					        """
 | 
				
			||||||
 | 
					        self.is_in_scope = None
 | 
				
			||||||
 | 
					        super(Scoped, self).__init__(*args, **kwargs)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    def in_scope(self):
 | 
				
			||||||
 | 
					        """
 | 
				
			||||||
 | 
					        Sets provider in "in scope" state.
 | 
				
			||||||
 | 
					        """
 | 
				
			||||||
 | 
					        self.is_in_scope = True
 | 
				
			||||||
 | 
					        self._reset_instance()
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    def out_of_scope(self):
 | 
				
			||||||
 | 
					        """
 | 
				
			||||||
 | 
					        Sets provider in "out of scope" state.
 | 
				
			||||||
 | 
					        """
 | 
				
			||||||
 | 
					        self.is_in_scope = False
 | 
				
			||||||
 | 
					        self._reset_instance()
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    def __call__(self, *args, **kwargs):
 | 
				
			||||||
 | 
					        """
 | 
				
			||||||
 | 
					        Returns provided instance.
 | 
				
			||||||
 | 
					        """
 | 
				
			||||||
 | 
					        if not self.is_in_scope:
 | 
				
			||||||
 | 
					            raise RuntimeError('Trying to provide {} while provider is not in scope'.format(self.provides))
 | 
				
			||||||
 | 
					        return super(Scoped, self).__call__(*args, **kwargs)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    def __enter__(self):
 | 
				
			||||||
 | 
					        """
 | 
				
			||||||
 | 
					        With __enter__() implementation. Makes provider to be in scope.
 | 
				
			||||||
 | 
					        """
 | 
				
			||||||
 | 
					        self.in_scope()
 | 
				
			||||||
 | 
					        return self
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    def __exit__(self, *_):
 | 
				
			||||||
 | 
					        """
 | 
				
			||||||
 | 
					        With __exit__() implementation. Makes provider to be out of scope.
 | 
				
			||||||
 | 
					        """
 | 
				
			||||||
 | 
					        self.out_of_scope()
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
class ExternalDependency(Provider):
 | 
					class ExternalDependency(Provider):
 | 
				
			||||||
    """
 | 
					    """
 | 
				
			||||||
| 
						 | 
					
 | 
				
			||||||
		Loading…
	
		Reference in New Issue
	
	Block a user