serverview/server.py at master · sbromfield/serverview · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
import redis
import tornado.ioloop
import tornado.web
import xml.etree.cElementTree as etree
import json

_host = "127.0.0.1"
_port = 6379
_db = 0
_r = None
enable_white_list = False
whitelist = ["131.94.19.25"]

class putdata(tornado.web.RequestHandler):
	#responses are sent in json
	def get(self):
		self.set_header("Content-Type", "application/json")
		#if we don't have all the variables send back an error
		try:
			name = self.get_argument("name")
			cpu = self.get_argument("cpu")
			network = self.get_argument("network")
			memory = self.get_argument("memory")
		except:
			self.write(json.dumps({"status": "error, invalid input"}))
			return
		#check if an ip is in the white list
		if enable_white_list:
			keepgoing = False
			for ip in whitelist:
				if ip == self.request.remote_ip:
					keepgoing = True
			if keepgoing:
				print "passed!"
			else:
				self.write(json.dumps({"status": "error, invalid user!"}))
				return
		#send data to redis and log results
		_r.set(name, memory+";"+cpu+";"+network+";")
		self.write(json.dumps({"status": "ok"}))
		print(name +": Memory " + memory+"; Cpu "+cpu+"; Network "+network+"; from ip " + self.request.remote_ip)

class getdata(tornado.web.RequestHandler):
	def get(self):
		keys = _r.keys()
		root = etree.Element("root")

		for key in keys:
			datastr = _r.get(key)
			data = datastr.split(";")

			if len(data) == 4:
				infonode = etree.SubElement(root, key)
				memorynode = etree.SubElement(infonode, "memory")
				memorynode.text = data[0]
				cpunode = etree.SubElement(infonode,"cpu")
				cpunode.text = data[1]
				networknode= etree.SubElement(infonode, "network")
				networknode.text = data[2]
			else:
				pass

		self.set_header("Content-Type", "text/xml")
		self.write(etree.tostring(root))
		print(etree.tostring(root))

application = tornado.web.Application([
	(r"/", putdata),
	(r"/getdata", getdata),
])

def connection():
	_r = redis.StrictRedis(host=_host, port=_port, db=_db)
	return _r

if __name__ == '__main__':
	_r = connection()
	application.listen(666)
	tornado.ioloop.IOLoop.instance().start()