[enh] category support

This commit is contained in:
asciimoo 2013-10-17 21:06:28 +02:00
parent 1aff439931
commit f76d80de54
3 changed files with 25 additions and 8 deletions

View file

@ -27,15 +27,23 @@ engine_dir = dirname(realpath(__file__))
engines = {}
categories = {'general': []}
for filename in listdir(engine_dir):
modname = splitext(filename)[0]
if filename.startswith('_') or not filename.endswith('.py'):
continue
filepath = join(engine_dir, filename)
engine = load_source(modname, filepath)
engine.name = modname
if not hasattr(engine, 'request') or not hasattr(engine, 'response'):
continue
engines[modname] = engine
if not hasattr(engine, 'categories'):
categories['general'].append(engine)
else:
for category_name in engine.categories:
categories.setdefault(category_name, []).append(engine)
def default_request_params():
return {'method': 'GET', 'headers': {}, 'data': {}, 'url': ''}