def devices():
if request.method == "GET":
dev_list = []
print(connected_devices)
mibands = copy.deepcopy(tmp_mibands)
for idx, mb in enumerate(mibands.keys()):
dev_id = mb2db.get_device_id(cnxn_string, mb)
dev_user = mb2db.get_device_user(cnxn_string, dev_id)
device = mb2db.get_device_by_id(cnxn_string, dev_id)
battery = -1
if device:
battery = device.bateria
username = (dev_user.nombre + " " +
dev_user.apellidos) if dev_user else "Unregistered"
dev_dict = {
"address": mb,
"signal": mibands[mibands.keys()[idx]].rssi,
"registered": False,
"connected": False,
"dev_id": dev_id,
"user_name": username,
"battery": battery,
"strikes": strikes[mibands[mibands.keys()[idx]].addr.upper()]
}
if mb2db.is_device_registered(cnxn_string, mb):
dev_dict["registered"] = True
if mb in connected_devices.keys():
dev_dict["connected"] = True
dev_list += [dev_dict]
print json.dumps(dev_list)
return json.dumps(dev_list)
elif request.method == "POST":
addr = request.form["address"].upper()
if mb2db.is_device_registered(cnxn_string, addr):
abort(403)
else:
try:
strikes[addr] = -9999
mb2 = MiBand2(addr, initialize=True)
mb2.cleanAlarms()
dev_id = mb2db.register_device(cnxn_string, mb2.addr)
mb2db.delete_all_alarms(cnxn_string, dev_id)
mb2db.update_battery(cnxn_string, mb2.addr,
mb2.battery_info['level'])
# Device stays connected after initialize, but we don't want that
mb2.disconnect()
strikes[addr] = 0
return json.dumps({"dev_id": dev_id, "registered": True})
except BTLEException as e:
print(
"There was a problem registering this MiBand2, try again later"
)
print e
abort(500)
except BTLEException.DISCONNECTED as d:
print("Device disconnected, removing from connected devices")
del connected_devices[addr]
del mb2
abort(500)
def do_activity(self, params):
try:
l = int(params)
except ValueError:
print "*** argument should be number"
return
except IndexError:
print "*** activity 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 ((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():
q.put(self.mibands.keys()[dev_id])
q.join()
else:
print(
"MiBand2 should be connected before fetching activity data"
)
else:
print("MiBand2 should be registered before fetching activity data")
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")
def do_reboot(self, params):
try:
dev_id = int(params)
except ValueError:
print "*** arguments should be numbers"
return
except IndexError:
print "*** alert takes at least one parameter"
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():
try:
mb2 = connected_devices[self.mibands.keys()[dev_id]]
mb2.reboot()
except BTLEException:
print(
"There was a problem rebooting this MiBand2, try again later"
)
else:
print("That MiBand2 is not connected!")
else:
print("That MiBand2 is not registered")
def do_devices(self, line):
tmp_mibands = copy.deepcopy(self.scd.tmp_devices)
self.mibands = {k: v["device"] for k, v in tmp_mibands.items()}
for idx, mb in enumerate(self.mibands.keys()):
name = "Someone"
uid = 0
udata = None
if args.mode == "db":
udata = mb2db.get_user_data(
mb2db.cnxn, mb2db.get_device_user(mb2db.cnxn, mb))
else:
# TODO: User Data on local storage???
pass
if udata:
name = udata["alias"]
uid = udata["id"]
str = "[%s]%10s's MB2 <U:%05d> (%s) %sdB " % (
idx, name, uid, mb,
self.mibands[self.mibands.keys()[idx]].rssi)
if (args.mode == "db" and mb2db.is_device_registered(
mb2db.cnxn, mb)) or (args.mode == "json"
and mb in self.registered_devices):
str += "[R]"
if mb in connected_devices:
mb2 = connected_devices[mb]
if args.mode == "db":
mb2db.update_battery(mb2db.cnxn, mb2.addr,
mb2.battery_info['level'])
str += "[C] [B:{0:03d}%]".format(mb2.battery_info["level"])
print str
def do_unregister(self, params):
try:
l = int(params)
except ValueError:
print "*** argument should be number"
return
except IndexError:
print "*** unregister 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 ((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 not self.mibands.keys()[dev_id] in connected_devices.values():
try:
if args.mode == "db":
mb2db.unregister_device(mb2db.cnxn, self.mibands.keys()[dev_id])
mb2db.delete_all_alarms(mb2db.cnxn, self.mibands.keys()[dev_id])
else:
self.registered_devices.remove(self.mibands.keys()[dev_id])
print("MiBand2 unregistered!")
except BTLEException:
print("There was a problem unregistering this MiBand2, try again later")
else:
print("Disconnect the miBand2 first!")
else:
print("That MiBand2 is not registered")
def do_reluser(self, params):
if args.mode == "db":
try:
l = params.split()
dev_id = int(l[0])
user_id = int(l[1])
except ValueError:
print "*** argument should be number"
return
except IndexError:
print "*** reluser takes at least one parameter"
return
if dev_id >= len(self.mibands.keys()):
print "*** device not in the device list"
return
udata = mb2db.get_user_data(mb2db.cnxn, user_id)
if udata:
if mb2db.is_device_registered(mb2db.cnxn, self.mibands.keys()[dev_id]):
if self.mibands.keys()[dev_id] in connected_devices.keys():
mb2 = connected_devices[self.mibands.keys()[dev_id]]
if mb2db.release_device_user(mb2db.cnxn, mb2.addr, user_id):
print "MiBand Released from user"
else:
print "There was a problem releasing this MiBand"
else:
print("MiBand2 should be connected before releasing user data")
else:
print("MiBand2 should be registered before releasing user data")
else:
print("*** user with id %s doesn't exist" % user_id)
else:
# TODO: If storage, release properly
print("This operation is only available for DB mode")
def do_alert(self, params):
l = params.split()
if len(l)!=2:
print "*** invalid number of arguments"
return
try:
l = [int(i) for i in l]
except ValueError:
print "*** arguments should be numbers"
return
except IndexError:
print "*** alert takes at least one parameter"
return
dev_id = int(l[0])
if dev_id >= len(self.mibands.keys()):
print "*** device not in the device list"
return
alert_int = int(l[1])
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():
try:
mb2 = connected_devices[self.mibands.keys()[dev_id]]
data = struct.pack('B', alert_int)
mb2.send_alert(data)
print "Sending Notification: " + binascii.hexlify(data)
except BTLEException:
print("There was a problem alerting this MiBand2, try again later")
else:
print("That MiBand2 is not connected!")
else:
print("That MiBand2 is not registered")
def do_setuser(self, params):
try:
l = params.split()
dev_id = int(l[0])
if args.mode == "db":
user_id = int(l[1])
else:
# TODO: Not persisted
user_alias = l[1]
if l[2] == "M":
user_gender = 0
elif l[2] == "F":
user_gender = 1
else:
user_gender = 2
user_bd_year = int(l[3])
user_bd_month = int(l[4])
user_bd_day = 0
user_weight = float(l[5])
user_height = int(l[6])
position = None
if l[2] == "left":
position = (0, "left")
elif l[2] == "right":
position = (1, "right")
else:
print("*** only left and right supported")
return
except ValueError:
print "*** argument should be number"
return
except IndexError:
print "*** setuser takes at least one parameter"
return
if dev_id >= len(self.mibands.keys()):
print "*** device not in the device list"
return
if args.mode == "db":
udata = mb2db.get_user_data(mb2db.cnxn, user_id)
if udata or args.mode == "json":
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":
if mb2db.set_device_user(mb2db.cnxn, mb2.addr, user_id, position[0]):
mb2.setUserInfo(udata["alias"], udata["sex"], udata["height"], udata["weight"], udata["birth"])
else:
mb2.setUserInfo(user_alias, user_gender, user_height, user_weight, (user_bd_year, user_bd_month, user_bd_day))
mb2.setWearLocation(position[1])
else:
print("MiBand2 should be connected before setting user data")
else:
print("MiBand2 should be registered before setting user data")
else:
print("*** user with id %s doesn't exist" % user_id)
def configure_miband(self, dev_id, preset):
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():
try:
mb2 = connected_devices[self.mibands.keys()[dev_id]]
print("Configuring MiBand to [%s] presets" % preset)
if config.has_option(preset, "MonitorHRSleep"):
mb2.monitorHeartRateSleep(config.getint(preset, "MonitorHRSleep"))
if config.has_option(preset, "MonitorHRInterval"):
mb2.setMonitorHeartRateInterval(config.getint(preset, "MonitorHRInterval"))
if config.has_option(preset, "DisplayTimeFormat"):
mb2.setDisplayTimeFormat(config.get(preset, "DisplayTimeFormat"))
if config.has_option(preset, "DisplayTimeHours"):
mb2.setDisplayTimeHours(config.getint(preset, "DisplayTimeHours"))
if config.has_option(preset, "DistanceUnit"):
mb2.setDistanceUnit(config.get(preset, "DistanceUnit"))
if config.has_option(preset, "LiftWristActivate"):
mb2.setLiftWristToActivate(config.getint(preset, "LiftWristActivate"))
if config.has_option(preset, "RotateWristSwitch"):
mb2.setRotateWristToSwitchInfo(config.getint(preset, "RotateWristSwitch"))
if config.has_option(preset, "DisplayItems"):
disp = [x.strip() for x in config.get(preset, 'DisplayItems').split(',')]
steps = True if 'steps' in disp else False
distance = True if 'distance' in disp else False
calories = True if 'calories' in disp else False
heartrate = True if 'heartrate' in disp else False
battery = True if 'battery' in disp else False
mb2.setDisplayItems(steps=steps, distance=distance, calories=calories, heartrate=heartrate, battery=battery)
if config.has_option(preset, "DoNotDisturb"):
enableLift = config.getint(preset, "DoNotDisturbLift") if config.has_option(preset, "DoNotDisturbLift") else 1
mb2.setDoNotDisturb(config.get(preset, "DoNotDisturb"), enableLift=enableLift)
if config.has_option(preset, "InactivityWarnings"):
start = config.getint(preset, "InactivityWarningsStart") if config.has_option(preset, "InactivityWarningsStart") else 8
end = config.getint(preset, "InactivityWarningsEnd") if config.has_option(preset, "InactivityWarningsEnd") else 19
threshold = config.getint(preset, "InactivityWarningsThresholdHours") if config.has_option(preset, "InactivityWarningsThresholdHours") else 1
mb2.setInactivityWarnings(config.getint(preset, "InactivityWarnings"), threshold=threshold*60, start=(start, 0), end=(end, 0))
if config.has_option(preset, "DisplayCaller"):
mb2.setDisplayCaller(config.getint(preset, "DisplayCaller"))
except BTLEException as e:
print("There was a problem configuring this MiBand2, try again later")
print e
else:
print("That MiBand2 is not connected, please connect it before configuring.")
else:
print("That MiBand2 is not registered, please register it before configuring.")
def do_register(self, params):
try:
l = int(params)
except ValueError:
print "*** argument should be number"
return
except IndexError:
print "*** register 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 ((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):
print("That MiBand2 is already registered")
else:
mb2 = None
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)
mb2.cleanAlarms()
if args.mode == "db":
mb2db.delete_all_alarms(mb2db.cnxn, mb2.addr)
mb2db.register_device(mb2db.cnxn, mb2.addr)
mb2db.update_battery(mb2db.cnxn, mb2.addr,
mb2.battery_info['level'])
else:
self.registered_devices += [mb2.addr]
# Device stays connected after initialize, but we don't want that
mb2.disconnect()
except BTLEException as e:
print(
"There was a problem registering this MiBand2, try again later"
)
print e
mb2.disconnect
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)
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")