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(),