#!/usr/bin/python

"""

Purpose: subscribing to AstroPiOTA and storing data on the Tangle

"""

import paho.mqtt.client as mqtt

import time

import json

from iota import Iota

from iota import ProposedTransaction

from iota import Address

from iota import Tag

from iota import TryteString

from json import load

def on_connect(client, userdata, flags, rc):

""" reporting IoT device connection """

try:

m = "Connected flags " + str(flags) + "\nResult code " + str(rc) + "\nClient_id " + str(client)

print(m)

print("\n")

except e:

print("Hmmm I couldn't report the IoT connection: ", e)

def on_message(client, userdata, msg):

""" receiving data"""

try:

sensors = msg.payload

sensors = json.loads(sensors.decode('utf-8'))

except e:

print("Check the message: ",e)

# this format stores data in CSV format in AstroPiOTA.log

logfile = open("AstroPiOTA.csv", "a")

print(str(sensors["timestamp"]),",",str(sensors["lng"]),",",\

str(sensors["lat"]),",",str(sensors["device_name"]),",",str(sensors["temperature"]),",",\

str(sensors["humidity"]),",",str(sensors["pressure"]),",",str(sensors["pitch"]),",",\

str(sensors["roll"]),",",str(sensors["yaw"]),",",str(sensors["x"]),",",\

str(sensors["y"]),",",str(sensors["z"]),",",str(sensors["device_owner"]),",",str(sensors["city"]),file=logfile)

logfile.close()

# this prints the AstroPiOTA data message

print("\nTimestamp: ", str(sensors["timestamp"]))

print("Device: ", sensors["device_name"])

print("Device owner email: ", sensors["device_owner"])

print("Device location: ", sensors["city"], " at longitude: ", sensors["lng"], " and latitude: ", sensors["lat"])

print("Temperature: ", sensors["temperature"])

print("Humidity: ", sensors["humidity"])

print("Pressure: ", sensors["pressure"])

print("Pitch: ", sensors["pitch"])

print("Roll: ", sensors["roll"])

print("Yaw: ", sensors["yaw"])

print("Accelerometer x: ", sensors["x"])

print("Accelerometer y: ", sensors["y"])

print("Accelerometer z: ", sensors["z"])

api = Iota('https://nodes.devnet.iota.org:443')

address = '999999999999999999999999999999999999999999999999999999999999999999999999999999999'

tx = ProposedTransaction(

address=Address(address),

#message=TryteString.from_unicode(sensors),

message=TryteString.from_unicode(json.dumps(sensors)),

tag=Tag('ASTROPIOTAIIIDEMO'),

value=0

)

print(tx)

try:

tx = api.prepare_transfer(transfers=[tx])

except:

print("PREPARE EXCEPTION",tx)

try:

result = api.send_trytes(tx['trytes'], depth=3, min_weight_magnitude=9)

except:

print("EXCEPTION", result)

def test_sub():

'''

Broker address: 18.217.227.236 (​ ec2-18-217-227-236.us-east-2.compute.amazonaws.com)

Broker port: 1883

account/pw: username/password that you obtain from the marketplace

topic: the name of the product that you purchased from the marketplace

'''

topic = "AstroPiOTA Weather Station"

account = 'YOUR-USERNAME'

pw = 'YOUR-PASSWORD'

sub_client = mqtt.Client(account)

sub_client.on_connect = on_connect

sub_client.on_message = on_message

sub_client.username_pw_set(account, pw)

sub_client.connect('18.217.227.236', 1883, 60) #connect to broker

sub_client.subscribe(topic)

rc = sub_client.loop_forever()

time.sleep(1)

print("Return code", rc)

if __name__ == '__main__':

# subscribing to AstroPiOTA data

try:

test_sub()

except KeyboardInterrupt:

rc = sub_client.loop_stop()

print("\nI'm stopping now")