WIP - initial code, just trying to make a POST request work

This commit is contained in:
Shay Nehmad 2019-05-30 09:17:24 +03:00
parent fd36118c0e
commit 026cb41064
2 changed files with 42 additions and 1 deletions

View File

@ -1,4 +1,7 @@
import json
import flask_restful import flask_restful
from flask import request
from monkey_island.cc.auth import jwt_required from monkey_island.cc.auth import jwt_required
from monkey_island.cc.services.edge import EdgeService from monkey_island.cc.services.edge import EdgeService
@ -28,4 +31,32 @@ class NetMap(flask_restful.Resource):
"edges": edges "edges": edges
} }
@jwt_required()
def post(self, **kw):
post_data = json.loads(request.data)
print(post_data)
monkeys = [NodeService.monkey_to_net_node(x) for x in mongo.db.monkey.find({})]
nodes = [NodeService.node_to_net_node(x) for x in mongo.db.node.find({})]
edges = [EdgeService.edge_to_net_edge(x) for x in mongo.db.edge.find({})]
if NodeService.get_monkey_island_monkey() is None:
monkey_island = [NodeService.get_monkey_island_pseudo_net_node()]
edges += EdgeService.get_monkey_island_pseudo_edges()
else:
monkey_island = []
edges += EdgeService.get_infected_monkey_island_pseudo_edges()
all_nodes = monkeys + nodes + monkey_island
def filter_linux(machine):
if machine["os"] == "linux":
return False
return True
all_nodes = filter(filter_linux, all_nodes)
return \
{
"nodes": all_nodes,
"edges": edges
}

View File

@ -103,7 +103,17 @@ class ReportPageComponent extends AuthComponent {
}; };
updateMapFromServer = () => { updateMapFromServer = () => {
this.authFetch('/api/netmap') this.authFetch('/api/netmap', {
method: 'POST',
headers: {
'Accept': 'application/json',
'Content-Type': 'application/json',
},
body: JSON.stringify({
firstParam: 'yourValue',
secondParam: 'yourOtherValue',
})
})
.then(res => res.json()) .then(res => res.json())
.then(res => { .then(res => {
res.edges.forEach(edge => { res.edges.forEach(edge => {