rasolar/pyweb.py

from bottle import get,post,request,Bottle,run,template
import threading,time,json,zlib,gnupg,socket,psutil
from queue import Queue


measdata={}

_HASH="hash"
_SIGNEDGPG="signed_gpg"
_PAYLOAD="payload"
_MEASURES="measure"
_BEGINSIGNATURE="-----BEGIN PGP SIGNATURE-----"
_BEGINMESSAGE="-----BEGIN PGP SIGNED MESSAGE-----"
_BEGINHASH="Hash:"

def sql_insert(q):
	while True:
		if q.empty():
			time.sleep(0.1)
			print("ping"+str(time.time()))
		else:
			try:
				indata=q.get()
				if indata is not None:
					q.task_done()
			except Exception as e:
				print("Error during queuing")
				print(e)
			else:
				print("Queue:")
				print(indata)
				if indata['hash'] in measdata:
					for i in indata['payload']['measures']:
						measdata[indata['hash']]['measures'][i]=indata['payload']['measures'][i]
				else:
					measdata[indata['hash']]=indata['payload']
				print(measdata)
					

app=Bottle()

@app.get('/')
def approot():
	return template('main.tpl',server=socket.gethostname(),cpupercent=psutil.cpu_percent(),measdata=measdata)

@app.post('/data/<hash_id:int>')
def dataimport(hash_id):
	print(hash_id)
	timestart=time.time()
	try:
		json_in=json.loads(request.json)
		print(json_in)
	except:
		print("no json")
	else:
		if _HASH in json_in:
			if int(json_in[_HASH]) == hash_id:
				print("correct id")
				bcorrect=False
				if _PAYLOAD in json_in:
					if _MEASURES in json_in[_PAYLOAD]:
						q.put(json_in,block=False)
						bcorrect=True
				if _SIGNEDGPG in json_in:
					if gpg.verify(json_in[_SIGNEDGPG]):
						signed_in=json_in[_SIGNEDGPG].split("\n")
						signed_in[signed_in.index(_BEGINSIGNATURE):]=""
						del signed_in[signed_in.index(_BEGINMESSAGE)]
						del signed_in[signed_in.index("")]
						for h in signed_in:
							if _BEGINHASH in h:
								del signed_in[signed_in.index(h)]
						if len(signed_in)>0:
							print(time.time()-timestart)
							q.put(json.loads(signed_in[0]),block=False)
						else:
							print("malformed signed packet")
							print(json_in)
					else:
						print("could not verify gpg signature")
						print(json_in)
			else:
				print("wrong id")
		else:
			print("json has no hash field")
			print(json_in)



gpg=gnupg.GPG()

q=Queue(maxsize=0)

sql_worker=threading.Thread(target=sql_insert,args=(q,))
sql_worker.setDaemon(True)
sql_worker.start()


run(app,host="localhost",port=8080)