2013-09-03 20:51:11 +00:00
|
|
|
#!/usr/bin/env python
|
2013-09-20 18:50:51 +00:00
|
|
|
# -*- coding: utf-8 -*-
|
2013-09-03 20:51:11 +00:00
|
|
|
|
|
|
|
"""
|
|
|
|
SleekXMPP: The Sleek XMPP Library
|
|
|
|
Implementation of xeps for Internet of Things
|
|
|
|
http://wiki.xmpp.org/web/Tech_pages/IoT_systems
|
|
|
|
Copyright (C) 2013 Sustainable Innovation, Joachim.lindborg@sust.se
|
|
|
|
This file is part of SleekXMPP.
|
|
|
|
|
|
|
|
See the file LICENSE for copying permission.
|
|
|
|
"""
|
|
|
|
|
2015-05-13 21:24:06 +00:00
|
|
|
import getpass
|
2013-09-03 20:51:11 +00:00
|
|
|
import logging
|
2015-05-13 21:24:06 +00:00
|
|
|
import sys
|
2013-09-03 20:51:11 +00:00
|
|
|
|
|
|
|
from optparse import OptionParser
|
|
|
|
from urllib import urlopen
|
|
|
|
|
|
|
|
import sleekxmpp
|
|
|
|
# Python versions before 3.0 do not use UTF-8 encoding
|
|
|
|
# by default. To ensure that Unicode is handled properly
|
|
|
|
# throughout SleekXMPP, we will set the default encoding
|
|
|
|
# ourselves to UTF-8.
|
|
|
|
if sys.version_info < (3, 0):
|
|
|
|
from sleekxmpp.util.misc_ops import setdefaultencoding
|
|
|
|
setdefaultencoding('utf8')
|
|
|
|
else:
|
|
|
|
raw_input = input
|
2013-09-20 18:50:51 +00:00
|
|
|
|
2013-09-03 20:51:11 +00:00
|
|
|
from sleekxmpp.plugins.xep_0323.device import Device
|
|
|
|
|
|
|
|
class IoT_TestDevice(sleekxmpp.ClientXMPP):
|
|
|
|
|
|
|
|
"""
|
|
|
|
A simple IoT device that can act as server or client
|
|
|
|
"""
|
|
|
|
def __init__(self, jid, password):
|
|
|
|
sleekxmpp.ClientXMPP.__init__(self, jid, password)
|
|
|
|
self.add_event_handler("session_start", self.session_start)
|
|
|
|
self.add_event_handler("message", self.message)
|
|
|
|
self.device=None
|
|
|
|
self.releaseMe=False
|
|
|
|
self.beServer=True
|
|
|
|
self.clientJID=None
|
|
|
|
|
|
|
|
def datacallback(self,from_jid,result,nodeId=None,timestamp=None,fields=None,error_msg=None):
|
|
|
|
"""
|
|
|
|
This method will be called when you ask another IoT device for data with the xep_0323
|
|
|
|
se script below for the registration of the callback
|
|
|
|
"""
|
|
|
|
logging.debug("we got data %s from %s",str(result),from_jid)
|
|
|
|
|
|
|
|
def beClientOrServer(self,server=True,clientJID=None ):
|
|
|
|
if server:
|
|
|
|
self.beServer=True
|
|
|
|
self.clientJID=None
|
|
|
|
else:
|
|
|
|
self.beServer=False
|
|
|
|
self.clientJID=clientJID
|
2013-09-20 18:50:51 +00:00
|
|
|
|
2013-09-03 20:51:11 +00:00
|
|
|
def testForRelease(self):
|
|
|
|
# todo thread safe
|
|
|
|
return self.releaseMe
|
|
|
|
|
|
|
|
def doReleaseMe(self):
|
|
|
|
# todo thread safe
|
|
|
|
self.releaseMe=True
|
2013-09-20 18:50:51 +00:00
|
|
|
|
2013-09-03 20:51:11 +00:00
|
|
|
def addDevice(self, device):
|
|
|
|
self.device=device
|
2013-09-20 18:50:51 +00:00
|
|
|
|
2013-09-03 20:51:11 +00:00
|
|
|
def session_start(self, event):
|
|
|
|
self.send_presence()
|
|
|
|
self.get_roster()
|
2013-09-20 18:50:51 +00:00
|
|
|
# tell your preffered friend that you are alive
|
2013-09-03 20:51:11 +00:00
|
|
|
self.send_message(mto='jocke@jabber.sust.se', mbody=self.boundjid.bare +' is now online use xep_323 stanza to talk to me')
|
|
|
|
|
|
|
|
if not(self.beServer):
|
|
|
|
session=self['xep_0323'].request_data(self.boundjid.full,self.clientJID,self.datacallback)
|
|
|
|
|
|
|
|
def message(self, msg):
|
|
|
|
if msg['type'] in ('chat', 'normal'):
|
|
|
|
logging.debug("got normal chat message" + str(msg))
|
|
|
|
ip=urlopen('http://icanhazip.com').read()
|
|
|
|
msg.reply("Hi I am " + self.boundjid.full + " and I am on IP " + ip).send()
|
|
|
|
else:
|
|
|
|
logging.debug("got unknown message type %s", str(msg['type']))
|
2013-09-20 18:50:51 +00:00
|
|
|
|
2013-09-03 20:51:11 +00:00
|
|
|
class TheDevice(Device):
|
|
|
|
"""
|
|
|
|
This is the actual device object that you will use to get information from your real hardware
|
|
|
|
You will be called in the refresh method when someone is requesting information from you
|
|
|
|
"""
|
|
|
|
def __init__(self,nodeId):
|
|
|
|
Device.__init__(self,nodeId)
|
|
|
|
self.counter=0
|
|
|
|
|
|
|
|
def refresh(self,fields):
|
|
|
|
"""
|
|
|
|
the implementation of the refresh method
|
|
|
|
"""
|
|
|
|
self._set_momentary_timestamp(self._get_timestamp())
|
|
|
|
self.counter+=self.counter
|
2013-09-20 18:50:51 +00:00
|
|
|
self._add_field_momentary_data(self, "Temperature", self.counter)
|
|
|
|
|
2013-09-03 20:51:11 +00:00
|
|
|
if __name__ == '__main__':
|
|
|
|
|
|
|
|
# Setup the command line arguments.
|
|
|
|
#
|
|
|
|
# This script can act both as
|
|
|
|
# "server" an IoT device that can provide sensorinformation
|
|
|
|
# python IoT_TestDevice.py -j "serverjid@yourdomain.com" -p "password" -n "TestIoT" --debug
|
|
|
|
#
|
|
|
|
# "client" an IoT device or other party that would like to get data from another device
|
2013-09-20 18:50:51 +00:00
|
|
|
|
2013-09-03 20:51:11 +00:00
|
|
|
optp = OptionParser()
|
|
|
|
|
|
|
|
# Output verbosity options.
|
|
|
|
optp.add_option('-q', '--quiet', help='set logging to ERROR',
|
|
|
|
action='store_const', dest='loglevel',
|
|
|
|
const=logging.ERROR, default=logging.INFO)
|
|
|
|
optp.add_option('-d', '--debug', help='set logging to DEBUG',
|
|
|
|
action='store_const', dest='loglevel',
|
|
|
|
const=logging.DEBUG, default=logging.INFO)
|
|
|
|
optp.add_option('-v', '--verbose', help='set logging to COMM',
|
|
|
|
action='store_const', dest='loglevel',
|
|
|
|
const=5, default=logging.INFO)
|
|
|
|
optp.add_option('-t', '--pingto', help='set jid to ping',
|
|
|
|
action='store', type='string', dest='pingjid',
|
|
|
|
default=None)
|
|
|
|
|
|
|
|
# JID and password options.
|
|
|
|
optp.add_option("-j", "--jid", dest="jid",
|
|
|
|
help="JID to use")
|
|
|
|
optp.add_option("-p", "--password", dest="password",
|
|
|
|
help="password to use")
|
|
|
|
|
|
|
|
# IoT test
|
|
|
|
optp.add_option("-c", "--sensorjid", dest="sensorjid",
|
|
|
|
help="Another device to call for data on", default=None)
|
|
|
|
optp.add_option("-n", "--nodeid", dest="nodeid",
|
|
|
|
help="I am a device get ready to be called", default=None)
|
2013-09-20 18:50:51 +00:00
|
|
|
|
2013-09-03 20:51:11 +00:00
|
|
|
opts, args = optp.parse_args()
|
|
|
|
|
|
|
|
# Setup logging.
|
|
|
|
logging.basicConfig(level=opts.loglevel,
|
|
|
|
format='%(levelname)-8s %(message)s')
|
|
|
|
|
|
|
|
if opts.jid is None:
|
|
|
|
opts.jid = raw_input("Username: ")
|
|
|
|
if opts.password is None:
|
|
|
|
opts.password = getpass.getpass("Password: ")
|
2013-09-20 18:50:51 +00:00
|
|
|
|
2013-09-03 20:51:11 +00:00
|
|
|
|
|
|
|
xmpp = IoT_TestDevice(opts.jid,opts.password)
|
|
|
|
xmpp.register_plugin('xep_0030')
|
|
|
|
#xmpp['xep_0030'].add_feature(feature='urn:xmpp:iot:sensordata',
|
|
|
|
# node=None,
|
|
|
|
# jid=None)
|
|
|
|
xmpp.register_plugin('xep_0323')
|
|
|
|
xmpp.register_plugin('xep_0325')
|
|
|
|
|
|
|
|
if opts.nodeid:
|
|
|
|
|
|
|
|
# xmpp['xep_0030'].add_feature(feature='urn:xmpp:sn',
|
|
|
|
# node=opts.nodeid,
|
|
|
|
# jid=xmpp.boundjid.full)
|
|
|
|
|
2014-08-17 22:52:24 +00:00
|
|
|
myDevice = TheDevice(opts.nodeid)
|
2013-09-03 20:51:11 +00:00
|
|
|
# myDevice._add_field(name="Relay", typename="numeric", unit="Bool");
|
2014-08-17 22:52:24 +00:00
|
|
|
myDevice._add_field(name="Temperature", typename="numeric", unit="C")
|
2013-09-03 20:51:11 +00:00
|
|
|
myDevice._set_momentary_timestamp("2013-03-07T16:24:30")
|
2014-08-17 22:52:24 +00:00
|
|
|
myDevice._add_field_momentary_data("Temperature", "23.4", flags={"automaticReadout": "true"})
|
2013-09-20 18:50:51 +00:00
|
|
|
|
2014-08-17 22:52:24 +00:00
|
|
|
xmpp['xep_0323'].register_node(nodeId=opts.nodeid, device=myDevice, commTimeout=10)
|
2013-09-03 20:51:11 +00:00
|
|
|
xmpp.beClientOrServer(server=True)
|
|
|
|
while not(xmpp.testForRelease()):
|
|
|
|
xmpp.connect()
|
2013-09-20 18:50:51 +00:00
|
|
|
xmpp.process(block=True)
|
2013-09-03 20:51:11 +00:00
|
|
|
logging.debug("lost connection")
|
|
|
|
if opts.sensorjid:
|
|
|
|
logging.debug("will try to call another device for data")
|
|
|
|
xmpp.beClientOrServer(server=False,clientJID=opts.sensorjid)
|
|
|
|
xmpp.connect()
|
|
|
|
xmpp.process(block=True)
|
|
|
|
logging.debug("ready ending")
|
2013-09-20 18:50:51 +00:00
|
|
|
|
2013-09-03 20:51:11 +00:00
|
|
|
else:
|
|
|
|
print "noopp didn't happen"
|
|
|
|
|