From 1b8e3fe2e4374757c63e681de34bffa62e4219d0 Mon Sep 17 00:00:00 2001 From: facetoe Date: Sun, 1 Feb 2015 09:19:31 +0800 Subject: [PATCH] Update setting util to locate modules with protected settings in required tuple. --- setting_util.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/setting_util.py b/setting_util.py index b5c941c..97cf663 100755 --- a/setting_util.py +++ b/setting_util.py @@ -42,6 +42,16 @@ for module_name in modules: if any([hasattr(clazz, setting) for setting in protected_settings]): credential_modules[clazz.__name__]['credentials'] = list(set(protected_settings) & set(members)) credential_modules[clazz.__name__]['key'] = "%s.%s" % (clazz.__module__, clazz.__name__) + elif hasattr(clazz, 'required'): + protected = [] + required = getattr(clazz, 'required') + for setting in protected_settings: + if setting in required: + protected.append(setting) + if protected: + credential_modules[clazz.__name__]['credentials'] = protected + credential_modules[clazz.__name__]['key'] = "%s.%s" % (clazz.__module__, clazz.__name__) + except ImportError: continue