From 880c45cd05234dc68b22d1372aa24b8f42d9aa8a Mon Sep 17 00:00:00 2001 From: Johannes Findeisen Date: Sat, 23 Nov 2013 01:26:53 +0100 Subject: [PATCH] added the needs_arguments method to mail and mongodb task --- linspector/tasks/alert/mail.py | 3 +++ linspector/tasks/storage/mongodb.py | 3 +++ 2 files changed, 6 insertions(+) diff --git a/linspector/tasks/alert/mail.py b/linspector/tasks/alert/mail.py index 05e5077..74634dd 100644 --- a/linspector/tasks/alert/mail.py +++ b/linspector/tasks/alert/mail.py @@ -64,6 +64,9 @@ class MailTask(Task): if "password" in args: self.password = args["password"] + def needs_arguments(self): + return True + def execute(self, job_information): logger.debug("Executing Mail Task!") diff --git a/linspector/tasks/storage/mongodb.py b/linspector/tasks/storage/mongodb.py index 7ef9b67..a996fcb 100644 --- a/linspector/tasks/storage/mongodb.py +++ b/linspector/tasks/storage/mongodb.py @@ -55,6 +55,9 @@ class MongodbTask(Task): self.mongo_db = self.mongo_client[self.database] self.mongo_db_collection = self.mongo_db[self.collection] + def needs_arguments(self): + return True + def execute(self, job_information): data = {"timestamp": str(datetime.datetime.now()), "msg": job_information.get_response_message(),