forked from p15670423/monkey
30 lines
850 B
Python
30 lines
850 B
Python
|
import json
|
||
|
|
|
||
|
|
import flask_restful
|
||
|
|
from bson import ObjectId
|
||
|
|
from flask import request
|
||
|
|
|
||
|
|
from cc.database import mongo
|
||
|
|
from cc.services.log import LogService
|
||
|
|
from cc.services.node import NodeService
|
||
|
|
|
||
|
|
__author__ = "itay.mizeretz"
|
||
|
|
|
||
|
|
|
||
|
|
class Log(flask_restful.Resource):
|
||
|
|
def get(self):
|
||
|
|
monkey_id = request.args.get('id')
|
||
|
|
exists_monkey_id = request.args.get('exists')
|
||
|
|
if monkey_id:
|
||
|
|
return LogService.get_log_by_monkey_id(ObjectId(monkey_id))
|
||
|
|
else:
|
||
|
|
return LogService.log_exists(ObjectId(exists_monkey_id))
|
||
|
|
|
||
|
|
def post(self):
|
||
|
|
telemetry_json = json.loads(request.data)
|
||
|
|
|
||
|
|
monkey_id = NodeService.get_monkey_by_guid(telemetry_json['monkey_guid'])['_id']
|
||
|
|
log_id = LogService.add_log(monkey_id, telemetry_json['log'])
|
||
|
|
|
||
|
|
return mongo.db.log.find_one_or_404({"_id": log_id})
|