diff --git a/modules/expansion/sourcecache.py b/modules/expansion/sourcecache.py index 0466372..14d2ca5 100755 --- a/modules/expansion/sourcecache.py +++ b/modules/expansion/sourcecache.py @@ -4,13 +4,17 @@ from url_archiver import url_archiver misperrors = {'error': 'Error'} mispattributes = {'input': ['link'], 'output': ['link']} moduleinfo = {'version': '0.1', 'author': 'Alexandre Dulaunoy', 'description': 'Module to cache web pages of analysis reports, OSINT sources. The module returns a link of the cached page.'} -archive_path = '/tmp/' +moduleconfig = ['archivepath'] def handler(q=False): if q is False: return False request = json.loads(q) + if (request.get('config')): + archive_path = request['config']['archivepath'] + else: + archive_path = '/tmp/' if request.get('link'): tocache = request['link'] archiver = url_archiver.Archive(archive_path=archive_path) @@ -28,4 +32,5 @@ def introspection(): def version(): + moduleinfo['config'] = moduleconfig return moduleinfo