'''Disconnect cleanly on SIGTERM or SIGINT'''

mqttc.publish("/clients/" + client_id, "Offline")

mqttc.disconnect()

logging.info("Disconnected from broker; exiting on signal %d", signum)

'''Test whether string contains a number (leading/traling white-space is ok)'''

try:

float(s)

return True

except ValueError:

logging.info("Connection to broker: %s", paho.connack_string(rc) )

if rc==0:

mqttc.publish("/clients/" + client_id, "Online")

map = userdata['map']

for topic in map:

logging.info("Subscribing to topic %s" % topic)

sock = userdata['sock']

host = userdata['carbon_server']

port = userdata['carbon_port']

lines = []

now = int(time.time())

map = userdata['map']

# Find out how to handle the topic in this message: slurp through

# our map

for t in map:

if paho.topic_matches_sub(t, msg.topic):

# print "%s matches MAP(%s) => %s" % (msg.topic, t, map[t])

# Must we rename the received msg topic into a different

# name for Carbon? In any case, replace MQTT slashes (/)

# by Carbon periods (.)

(type, remap) = map[t]

if remap is None:

carbonkey = msg.topic.replace('/', '.')

else:

carbonkey = remap.replace('/', '.')

carbonkey=carbonkey.lstrip('.')

logging.debug("CARBONKEY is [%s]" % carbonkey)

try:

# Try to decode data as json

st = json.loads(msg.payload)

for k in st:

if is_number(st[k]):

lines.append("%s.%s %f %d" % (carbonkey, k, float(st[k]), now))

except:

# Try to decode as simple number

try:

lines.append("%s %f %d" % (carbonkey, float(msg.payload), now))

except ValueError:

logging.info("Topic %s contains payload [%s] as unknown data format" %

(msg.topic, msg.payload))

return

message = '\n'.join(lines) + '\n'

logging.debug("%s", message.strip())

if rc == 0:

logging.info("Clean disconnection")

else:

logging.info("Unexpected disconnect (%s); reconnecting in 5 seconds", paho.connack_string(rc) )

d = string.split(":")

host = d[0]

port = int(d[1]) if len(d)>1 else default_port

class LoadFromFile( argparse.Action ):

def __call__ (self, parser, namespace, values, option_string = None):

with values as f:

parser.parse_args(f.read().split(), namespace)

parser = argparse.ArgumentParser( fromfile_prefix_chars='@' )

parser.add_argument( "-c", "--config", type=open, action=LoadFromFile, help="Load config from file" )

parser.add_argument( "--mqtt", default="localhost:1883", type=urlparse.urlparse )

parser.add_argument( "--auth" )

parser.add_argument( "--carbon", default="127.0.0.1:2003" )

parser.add_argument( "-m", action="append", nargs="*", dest="map" )

parser.add_argument( "-v", action="store_true", default=False, help="Verbose logging", dest="verbose" )

parser.add_argument( "--logfile", help="Logging into file" )

args = parser.parse_args()

# configure logging

logging.basicConfig(format="[%(asctime)s] [%(levelname)s] [%(name)s] %(message)s", level=logging.DEBUG if args.verbose else logging.INFO, filename=args.logfile )

map = {}

for item in args.map:

type = item[0]

topic = item[1]

remap = None if len(item)==2 else item[3]

map[topic] = (type, remap)

try:

sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)

except:

sys.stderr.write("Can't create UDP socket\n")

sys.exit(1)

host = parse_host_port(args.carbon,2003)

userdata = {

'sock' : sock,

'carbon_server' : host[0],

'carbon_port' : host[1],

'map' : map,

}

global mqttc

mqttc = paho.Client(client_id, clean_session=True, userdata=userdata)

if args.mqtt.username!=None:

mqttc.username_pw_set(args.mqtt.username, args.mqtt.password )

mqttc.on_message = on_message

mqttc.on_connect = on_connect

mqttc.on_disconnect = on_disconnect

mqttc.on_subscribe = on_subscribe

mqttc.connect(args.mqtt.hostname, 1883 if args.mqtt.port==None else args.mqtt.port , 60)

signal.signal(signal.SIGTERM, cleanup)

signal.signal(signal.SIGINT, cleanup)