From d58ab345b6995a780ffef5bfb5a5b22003e65c92 Mon Sep 17 00:00:00 2001 From: schneefux Date: Sat, 24 May 2014 18:46:41 -0700 Subject: Filter modules intelligently --- client/brain.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/client/brain.py b/client/brain.py index 2e7607a..2babfdf 100644 --- a/client/brain.py +++ b/client/brain.py @@ -38,8 +38,6 @@ class Brain(object): def get_module_names(): module_names = [m.replace('.py', '') for m in listdir(folder) if m.endswith('.py')] - module_names.remove('__init__') - module_names.remove('app_utils') module_names = map(lambda s: folder + '.' + s, module_names) return module_names @@ -57,6 +55,7 @@ class Brain(object): return 0 modules = map(import_module, get_module_names()) + modules = filter(lambda m: hasattr(m, 'WORDS'), modules) modules.sort(key=get_module_priority, reverse=True) return modules -- cgit v1.3.1