From bab7265a90dd00bda1091e95530f2d31be43db9e Mon Sep 17 00:00:00 2001 From: Roman Mogilatov Date: Tue, 1 Sep 2015 00:36:26 +0300 Subject: [PATCH] Make minor import refactorings --- dependency_injector/catalog.py | 11 +++++------ dependency_injector/utils.py | 12 ++++++------ 2 files changed, 11 insertions(+), 12 deletions(-) diff --git a/dependency_injector/catalog.py b/dependency_injector/catalog.py index 033e60fc..e1de103a 100644 --- a/dependency_injector/catalog.py +++ b/dependency_injector/catalog.py @@ -1,7 +1,6 @@ """Catalog module.""" -from six import iteritems -from six import add_metaclass +import six from .errors import Error from .utils import is_provider @@ -15,7 +14,7 @@ class CatalogMetaClass(type): """Meta class factory.""" providers = dict() new_attributes = dict() - for name, value in iteritems(attributes): + for name, value in six.iteritems(attributes): if is_provider(value): providers[name] = value new_attributes[name] = value @@ -26,7 +25,7 @@ class CatalogMetaClass(type): return cls -@add_metaclass(CatalogMetaClass) +@six.add_metaclass(CatalogMetaClass) class AbstractCatalog(object): """Abstract providers catalog.""" @@ -54,7 +53,7 @@ class AbstractCatalog(object): def filter(cls, provider_type): """Return dict of providers, that are instance of provided type.""" return dict([(name, provider) - for name, provider in iteritems(cls.providers) + for name, provider in six.iteritems(cls.providers) if isinstance(provider, provider_type)]) @classmethod @@ -63,7 +62,7 @@ class AbstractCatalog(object): :type overriding: AbstractCatalog """ - for name, provider in iteritems(overriding.providers): + for name, provider in six.iteritems(overriding.providers): cls.providers[name].override(provider) diff --git a/dependency_injector/utils.py b/dependency_injector/utils.py index 0c0f35c4..f3ed3467 100644 --- a/dependency_injector/utils.py +++ b/dependency_injector/utils.py @@ -1,13 +1,13 @@ """Utils module.""" -from six import class_types +import six from .errors import Error def is_provider(instance): """Check if instance is provider instance.""" - return (not isinstance(instance, class_types) and + return (not isinstance(instance, six.class_types) and getattr(instance, '__IS_PROVIDER__', False) is True) @@ -21,7 +21,7 @@ def ensure_is_provider(instance): def is_injection(instance): """Check if instance is injection instance.""" - return (not isinstance(instance, class_types) and + return (not isinstance(instance, six.class_types) and getattr(instance, '__IS_INJECTION__', False) is True) @@ -35,19 +35,19 @@ def ensure_is_injection(instance): def is_kwarg_injection(instance): """Check if instance is keyword argument injection instance.""" - return (not isinstance(instance, class_types) and + return (not isinstance(instance, six.class_types) and getattr(instance, '__IS_KWARG_INJECTION__', False) is True) def is_attribute_injection(instance): """Check if instance is attribute injection instance.""" - return (not isinstance(instance, class_types) and + return (not isinstance(instance, six.class_types) and getattr(instance, '__IS_ATTRIBUTE_INJECTION__', False) is True) def is_method_injection(instance): """Check if instance is method injection instance.""" - return (not isinstance(instance, class_types) and + return (not isinstance(instance, six.class_types) and getattr(instance, '__IS_METHOD_INJECTION__', False) is True)