From b0773ca8b88cd9116e5f8e5e02f323c6670b57e9 Mon Sep 17 00:00:00 2001 From: Ilya Kreymer Date: Wed, 1 Apr 2015 17:03:36 -0700 Subject: [PATCH] pywb_init: ability to override DirectoryCollsLoader with custom class --- pywb/webapp/pywb_init.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/pywb/webapp/pywb_init.py b/pywb/webapp/pywb_init.py index 9f539533..f1ae0fea 100644 --- a/pywb/webapp/pywb_init.py +++ b/pywb/webapp/pywb_init.py @@ -252,8 +252,10 @@ def create_wb_router(passed_config={}): static_routes = config.get('static_routes', {}) + colls_loader_cls = config.get('colls_loader_cls', DirectoryCollsLoader) + # collections based on file system - dir_loader = DirectoryCollsLoader(config, static_routes) + dir_loader = colls_loader_cls(config, static_routes) collections.update(dir_loader()) if config.get('enable_memento', False):