70 lines
No EOL
2.1 KiB
Python
70 lines
No EOL
2.1 KiB
Python
"""
|
|
The MongoDB task
|
|
|
|
Copyright (c) 2011-2013 by Johannes Findeisen and Rafael Timmerberg
|
|
|
|
This file is part of Linspector (http://linspector.org).
|
|
|
|
Linspector is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU Affero General Public License as
|
|
published by the Free Software Foundation, either version 3 of the
|
|
License, or (at your option) any later version.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU Affero General Public License for more details.
|
|
|
|
You should have received a copy of the GNU Affero General Public License
|
|
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
"""
|
|
|
|
import datetime
|
|
from logging import getLogger
|
|
from pymongo import MongoClient
|
|
|
|
from linspector.tasks.task import Task
|
|
|
|
logger = getLogger(__name__)
|
|
|
|
|
|
class MongodbTask(Task):
|
|
def __init__(self, **kwargs):
|
|
super(MongodbTask, self).__init__(**kwargs)
|
|
|
|
args = self.get_arguments()
|
|
|
|
if "database" in args:
|
|
self.database = args["database"]
|
|
else:
|
|
raise Exception("There is no database set")
|
|
|
|
self.collection = "default"
|
|
if "collection" in args:
|
|
self.collection = args["collection"]
|
|
|
|
self.host = "localhost"
|
|
if "host" in args:
|
|
self.host = args["host"]
|
|
|
|
self.port = 27017
|
|
if "port" in args:
|
|
self.port = args["port"]
|
|
|
|
self.mongo_client = MongoClient(self.host, self.port)
|
|
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(),
|
|
"status": job_information.get_status(),
|
|
"job_id": job_information.get_job_id()}
|
|
self.mongo_db_collection.insert(data)
|
|
|
|
|
|
def create(kwargs):
|
|
return MongodbTask(**kwargs) |