#!/usr/bin/python
import ec3k
import signal
import time

want_stop = False

def handler(signum, frame):
        global want_stop
        print "Signal %d caught! Stopping..." % (signum,)
        want_stop = True

def main():
        signal.signal(signal.SIGTERM, handler)
        signal.signal(signal.SIGINT, handler)

        filehandlers = dict()

        def callback(state):
                id = "%04x" % state.id

                if not id in filehandlers:
                        file = "/var/run/ec3k/%s" % id
                        filehandlers[id] = open(file, "w")

                filehandlers[id].seek(0,0)
                filehandlers[id].write("%d,%s,%d,%d" % (int(time.time()),id, state.time_total, state.energy))

        my_ec3k = ec3k.EnergyCount3K(callback=callback, freq=868.185e6)
        my_ec3k.start()

        while not want_stop:
                time.sleep(60)
                #print "Noise level: %.1f dB" % (my_ec3k.noise_level,)

        my_ec3k.stop()

        for handle in filehandlers:
                filehandlers[handle].close()

if __name__ == '__main__':
        main()