Exemplo n.º 1
0
 def do_connect(self, params):
     try:
         l = int(params)
     except ValueError:
         print "*** argument should be number"
         return
     except IndexError:
         print "*** connect takes at least one parameter"
         return
     dev_id = l
     if dev_id >= len(self.mibands.keys()):
         print "*** device not in the device list"
         return
     if len(connected_devices.keys()) >= 5:
         print(
             "Can't connect to more than 5 devices at the same time, disconnect some"
         )
     else:
         if ((args.mode == "db"
              and mb2db.is_device_registered(mb2db.cnxn,
                                             self.mibands.keys()[dev_id]))
                 or args.mode == "json" and self.mibands.keys()[dev_id]
                 in self.registered_devices):
             if self.mibands.keys()[dev_id] in connected_devices.keys():
                 print("That MiBand2 is already connected")
             else:
                 try:
                     addr = self.mibands.keys()[dev_id]
                     self.scd.tmp_devices[addr]["strikes"] = -9999
                     if not addr in self.devices_keys.keys():
                         self.devices_keys[addr] = random_key()
                     mb2 = MiBand2(addr,
                                   self.devices_keys[addr],
                                   initialize=False)
                     connected_devices[self.mibands.keys()[dev_id]] = mb2
                     if args.mode == "db":
                         alarms = mb2db.get_device_alarms(
                             mb2db.cnxn, mb2.addr)
                         mb2db.update_battery(mb2db.cnxn, mb2.addr,
                                              mb2.battery_info['level'])
                     else:
                         if mb2.addr in self.devices_alarms.keys():
                             alarms = self.devices_alarms[mb2.addr]
                         else:
                             alarms = []
                     for a in alarms:
                         mb2.alarms += [
                             MiBand2Alarm(a["hour"],
                                          a["minute"],
                                          enabled=a["enabled"],
                                          repetitionMask=a["repetition"])
                         ]
                 except BTLEException as e:
                     print(
                         "There was a problem connecting to this MiBand2, try again later"
                     )
                     print e
         else:
             print(
                 "You have to register the MiBand2 before connecting to it")
Exemplo n.º 2
0
def device(dev_id):
    row = mb2db.get_device_by_id(cnxn_string, dev_id)
    if row:
        if request.method == "GET":
            connected = True if row.mac in connected_devices else False
            signal = 0
            mibands = copy.deepcopy(tmp_mibands)
            if row.mac in mibands.keys():
                signal = mibands[row.mac].rssi
            dev_user = mb2db.get_device_user(cnxn_string, dev_id)
            username = (dev_user.nombre + " " +
                        dev_user.apellidos) if dev_user else "Unregistered"
            detail_dict = {
                "dev_id": row.dispositivoId,
                "battery": row.bateria,
                "registered": row.registrado,
                "address": row.mac,
                "connected": connected,
                "signal": signal,
                "visible": (signal < 0),
                "user_name": username
            }
            return json.dumps(detail_dict)
        elif request.method == "PUT":
            if mb2db.is_device_registered(cnxn_string, row.mac):
                action = request.form.get("action")
                strikes[row.mac] = -9999
                if action == "connect" and row.mac not in connected_devices.keys(
                ):
                    try:
                        mb2 = MiBand2(row.mac, initialize=False)
                        connected_devices[row.mac] = mb2
                        alarms = mb2db.get_device_alarms(cnxn_string, mb2.addr)
                        mb2db.update_battery(cnxn_string, mb2.addr,
                                             mb2.battery_info['level'])
                        for a in alarms:
                            mb2.alarms += [
                                MiBand2Alarm(a["hour"],
                                             a["minute"],
                                             enabled=a["enabled"],
                                             repetitionMask=a["repetition"])
                            ]
                        return json.dumps({
                            "connected": True,
                            "dev_id": row.dispositivoId
                        }), 200
                    except BTLEException as e:
                        print(
                            "There was a problem (dis)connecting to this MiBand2, try again later"
                        )
                        print e
                        abort(500)
                    except BTLEException.DISCONNECTED as d:
                        print(
                            "Device disconnected, removing from connected devices"
                        )
                        del connected_devices[row.mac]
                        del mb2
                        abort(500)
                elif action == "disconnect" and row.mac in connected_devices.keys(
                ):
                    try:
                        mb2 = connected_devices[row.mac]
                        mb2.disconnect()
                        del connected_devices[row.mac]
                        del mb2
                        print("MiBand2 disconnected!")
                        return json.dumps({
                            "connected": False,
                            "dev_id": row.dispositivoId
                        }), 200
                    except BTLEException as e:
                        print(
                            "There was a problem disconnecting this MiBand2, try again later"
                        )
                        print e
                        abort(500)
                    except BTLEException.DISCONNECTED as d:
                        print(
                            "Device disconnected, removing from connected devices"
                        )
                        del connected_devices[row.mac]
                        del mb2
                        abort(500)
        elif request.method == "DELETE":
            # Just Unregister MiBand2
            if mb2db.is_device_registered(cnxn_string, row.mac):
                if not row.mac in connected_devices.keys():
                    try:
                        dev_id = mb2db.get_device_id(cnxn_string, row.mac)
                        mb2db.unregister_device(cnxn_string, dev_id)
                        mb2db.delete_all_alarms(cnxn_string, dev_id)
                        print("MiBand2 unregistered!")
                        return json.dumps({
                            "registered": False,
                            "dev_id": row.dispositivoId
                        }), 200
                    except BTLEException as e:
                        print(
                            "There was a problem unregistering this MiBand2, try again later"
                        )
                        print e
                        abort(500)
                    except BTLEException.DISCONNECTED as d:
                        print(
                            "Device disconnected, removing from connected devices"
                        )
                        del connected_devices[row.mac]
                        del mb2
                        abort(500)
        abort(403)
    else:
        abort(404)
Exemplo n.º 3
0
    def do_alarms(self, params):
        l = params.split()
        try:
            dev_id = int(l[0])
            command = "list"
            if len(l) > 1:
                command = l[1]
        except ValueError:
            print "*** argument 1 should be number"
            return
        except IndexError:
            print "*** alarms takes at least one parameter"
            return
        if command not in [
                'list', 'queue', 'set', 'toggle', 'toggleday', 'delete',
                'clear'
        ]:
            print "*** invalid alarm command, see help"
            return
        if dev_id >= len(self.mibands.keys()):
            print "*** device not in the device list"
            return
        if ((args.mode == "db"
             and mb2db.is_device_registered(mb2db.cnxn,
                                            self.mibands.keys()[dev_id]))
                or args.mode == "json"
                and self.mibands.keys()[dev_id] in self.registered_devices):
            if self.mibands.keys()[dev_id] in connected_devices.keys():
                mb2 = connected_devices[self.mibands.keys()[dev_id]]
                if args.mode == "db":
                    alarms = mb2db.get_device_alarms(
                        mb2db.cnxn,
                        self.mibands.keys()[dev_id])
                else:
                    if self.mibands.keys()[dev_id] in self.devices_alarms.keys(
                    ):
                        alarms = self.devices_alarms[self.mibands.keys()
                                                     [dev_id]]
                    else:
                        alarms = []
                if command == 'list':
                    if len(alarms) > 0:
                        for idx, a in enumerate(mb2.alarms):
                            print "[%s]" % idx + str(a)
                if command == 'clear':
                    if len(alarms) > 0:
                        mb2.cleanAlarms()
                        if args.mode == "db":
                            mb2db.delete_all_alarms(mb2db.cnxn, mb2.addr)
                        else:
                            self.devices_alarms[self.mibands.keys()
                                                [dev_id]] = []
                elif command == 'queue':
                    try:
                        hour, minute = map(lambda x: int(x), l[2].split(":"))
                        alarm_id = mb2.queueAlarm(hour, minute)
                        if args.mode == "db":
                            mb2db.set_alarm(mb2db.cnxn, mb2.addr,
                                            mb2.alarms[alarm_id], alarm_id)
                        else:
                            if len(alarms) > 0:
                                self.devices_alarms[self.mibands.keys()
                                                    [dev_id]] += [{
                                                        "enabled":
                                                        True,
                                                        "repetition":
                                                        128,
                                                        "hour":
                                                        hour,
                                                        "minute":
                                                        minute
                                                    }]
                            else:
                                self.devices_alarms[self.mibands.keys()
                                                    [dev_id]] = [{
                                                        "enabled":
                                                        True,
                                                        "repetition":
                                                        128,
                                                        "hour":
                                                        hour,
                                                        "minute":
                                                        minute
                                                    }]
                    except IndexError:
                        print "*** queue takes an hour parameter in format HH:MM"
                    except ValueError:
                        print "*** queue takes an hour parameter in format HH:MM"
                elif command == 'delete':
                    try:
                        alarm_id = int(l[2])
                        mb2.deleteAlarm(alarm_id)
                        if len(alarms) > 0:
                            if args.mode == "db":
                                mb2db.delete_alarm(mb2db.cnxn, mb2.addr,
                                                   alarm_id)
                            else:
                                del self.devices_alarms[self.mibands.keys()
                                                        [dev_id]][alarm_id]
                    except IndexError:
                        print "*** delete takes an alarm_id parameter"
                    except ValueError:
                        print "*** delete's alarm_id should be a number"
                elif command == 'toggle':
                    try:
                        alarm_id = int(l[2])
                        mb2.toggleAlarm(alarm_id)
                        if args.mode == "db":
                            mb2db.set_alarm(mb2db.cnxn, mb2.addr,
                                            mb2.alarms[alarm_id], alarm_id)
                        else:
                            self.devices_alarms[
                                self.mibands.keys()[dev_id]][alarm_id][
                                    "enabled"] = mb2.alarms[alarm_id].enabled
                    except IndexError:
                        print "*** toggle takes an alarm_id parameter"
                    except ValueError:
                        print "*** toggle's alarm_id should be a number"
                elif command == 'toggleday':
                    try:
                        alarm_id = int(l[2])
                        day_id = int(l[3])
                        if day_id not in range(1, 8):
                            print "*** day_id should be between 1 (Monday) and 7 (Sunday)"
                            return
                        else:
                            mb2.toggleAlarmDay(alarm_id, day_id - 1)
                            if args.mode == "db":
                                mb2db.set_alarm(mb2db.cnxn, mb2.addr,
                                                mb2.alarms[alarm_id], alarm_id)
                            else:
                                self.devices_alarms[
                                    self.mibands.keys()[dev_id]][alarm_id][
                                        "repetition"] = mb2.alarms[
                                            alarm_id].repetitionMask

                    except IndexError:
                        print "*** toggleday takes an alarm_id parameter and a day_id parameter (1-7)"
                    except ValueError:
                        print "*** toggleday's alarm_id and day_id should be both numbers"
                elif command == "set":
                    try:
                        alarm_id = int(l[2])
                        hour, minute = map(lambda x: int(x), l[3].split(":"))
                        mb2.changeAlarmTime(alarm_id, hour, minute)
                        if args.mode == "db":
                            mb2db.set_alarm(mb2db.cnxn, mb2.addr,
                                            mb2.alarms[alarm_id], alarm_id)
                        else:
                            self.devices_alarms[self.mibands.keys()[dev_id]][
                                alarm_id]["hour"] = mb2.alarms[alarm_id].hour
                            self.devices_alarms[
                                self.mibands.keys()[dev_id]][alarm_id][
                                    "minute"] = mb2.alarms[alarm_id].minute
                    except IndexError:
                        print "*** set takes an alarm_id parameter and an hour parameter in format HH:MM"
                    except ValueError:
                        print "*** toggleday's alarm_id and hour (HH:MM) should be both numbers"
            else:
                print(
                    "MiBand2 should be connected before viewing/changing alarms"
                )
        else:
            print(
                "MiBand2 should be registered before viewing/changing alarms")