130 lines
3.8 KiB
Python
130 lines
3.8 KiB
Python
#!/usr/bin/env python3
|
|
|
|
import urllib
|
|
import psycopg2
|
|
import websocket
|
|
# import xml.etree.ElementTree as ET
|
|
from http.server import BaseHTTPRequestHandler, HTTPServer
|
|
import re
|
|
import json
|
|
import pprint
|
|
|
|
# HTTPRequestHandler class
|
|
class testHTTPServer_RequestHandler(BaseHTTPRequestHandler):
|
|
# not used, just a sample
|
|
def do_GET(self):
|
|
# Send response status code
|
|
self.send_response(200)
|
|
|
|
# Send headers
|
|
self.send_header('Content-type','text/html')
|
|
self.end_headers()
|
|
|
|
# Send message back to client
|
|
message = "Hello world!"
|
|
# Write content as utf-8 data
|
|
self.wfile.write(bytes(message, "utf8"))
|
|
return
|
|
|
|
def do_POST(self):
|
|
length = int(self.headers['Content-Length'])
|
|
post_data = self.rfile.read(length).decode('utf-8')
|
|
|
|
# Print on stdout
|
|
print(post_data)
|
|
|
|
payload = self.dataToString(post_data)
|
|
deveui = self.dataToDevEUI(post_data)
|
|
print("{}:{}".format(deveui, payload))
|
|
|
|
# And insert into the db
|
|
self.insert_json("swisscom", post_data, payload, deveui)
|
|
|
|
# Sendo to dashboard
|
|
# self.to_dashboard(post_data)
|
|
|
|
def to_dashboard(self, data):
|
|
dev = self.devEUI(data)
|
|
text = self.payload(data)
|
|
|
|
ws = websocket.create_connection("wss://home-safety-visual.eu-gb.mybluemix.net/data")
|
|
ws.send("%s" % data)
|
|
ws.close()
|
|
|
|
ws = websocket.create_connection("wss://home-safety-visual.eu-gb.mybluemix.net/rawmessage")
|
|
ws.send("%s:%s" % (dev, text))
|
|
ws.close()
|
|
|
|
# Working lora node
|
|
if dev == "9CD90BB52B6A1D01":
|
|
try:
|
|
key, value = text.split("=")
|
|
print("Trying to send: %s:%s" % (key, value))
|
|
ws = websocket.create_connection("wss://home-safety-visual.eu-gb.mybluemix.net/%s" % (key))
|
|
ws.send("%s" % (value))
|
|
ws.close()
|
|
|
|
except ValueError:
|
|
print("Cannot split: %s" % (text))
|
|
|
|
|
|
def jsonToDict(self, data):
|
|
return json.loads(data)
|
|
|
|
def dictToPayload(self, thedict):
|
|
return thedict['DevEUI_uplink']['payload_hex']
|
|
|
|
def hexToString(self, myhex):
|
|
return bytes.fromhex(myhex).decode('utf-8')
|
|
|
|
def dataToString(self, data):
|
|
mydict = self.jsonToDict(data)
|
|
payload = self.dictToPayload(mydict)
|
|
return self.hexToString(payload)
|
|
|
|
def dataToDevEUI(self, data):
|
|
mydict = self.jsonToDict(data)
|
|
eui = mydict['DevEUI_uplink']['DevEUI']
|
|
return eui
|
|
|
|
def devEUI(self, data):
|
|
root = ET.fromstring(data)
|
|
return root[1].text
|
|
|
|
def payload_hex(self, data):
|
|
root = ET.fromstring(data)
|
|
return root[7].text
|
|
|
|
def payload(self, data):
|
|
myhex = self.payload_hex(data)
|
|
return bytes.fromhex(myhex).decode('utf-8')
|
|
|
|
def insert_json(self, provider, data, payload='', deveui=''):
|
|
try:
|
|
conn = psycopg2.connect("dbname=lorawan")
|
|
cursor = conn.cursor()
|
|
cursor.execute("insert into packets values (DEFAULT, DEFAULT, %s, %s, %s, %s)", (provider, data, payload, deveui))
|
|
cursor.connection.commit()
|
|
conn.close()
|
|
except Exception as e:
|
|
print("DB Insert failed: %s" % e)
|
|
|
|
|
|
def insert_xml(self, data):
|
|
try:
|
|
conn = psycopg2.connect("dbname=hackzurich")
|
|
cursor = conn.cursor()
|
|
cursor.execute("insert into data (packet) values (%s)", (data, ))
|
|
cursor.connection.commit()
|
|
conn.close()
|
|
except Exception as e:
|
|
print("DB Insert failed: %s" % e)
|
|
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
server_address = ('0.0.0.0', 8000)
|
|
httpd = HTTPServer(server_address, testHTTPServer_RequestHandler)
|
|
print('running server...')
|
|
httpd.serve_forever()
|