__init__.py

from .ble import BLEDevice
from .dte import DTE
from .ota_fw import OTAFW


class Scanner():
    def __init__(self):
        self._device = BLEDevice()

    def scan(self):
        return [x for x in self._device.scan() if (x.name and ('Linkit' in x.name or 'Horizon' in x.name))]


class Tracker():
    def __init__(self, address):
        self._device = BLEDevice()
        self._device.connect(address, 5)
        self._dte = DTE(self._device)
        self._otafw = OTAFW(self._device)
        self._map = {}

    def sync(self):
        a = self._dte.parmr()
        b = self._dte.statr()
        self._map = { **a, **b }

    def set(self, param_values):
        self._dte.parmw(param_values=param_values)

    def get(self, attr=None):
        return self._map[attr] if attr else self._map

    def get_attrs(self):
        return self._map.keys()

    def firmware_update(self, data, file_id=0, timeout=None):
        self._otafw.send_update_file(file_id, data, timeout)

    def paspw(self, json_file_data):
        self._dte.paspw(json_file_data)

    def dumpd(self, log_type):
        return self._dte.dumpd(log_type)

    def erase(self, log_type):
        return self._dte.erase(log_type)

    def factw(self):
        self._dte.factw()

    def rstvw(self, index):
        self._dte.rstvw(index)

    def rstbw(self):
        self._dte.rstbw()

    def deplw(self):
        self._dte.rstbw()

    def scalw(self, sensor, step, value=0):
        self._dte.scalw(sensor, step, value)

    def scalr(self, sensor, step):
        return self._dte.scalr(sensor, step)

    def argostx(self, mod, power, freq, size, tcxo):
        self._dte.argostx(mod, power, freq, size, tcxo)

    def poll(self, key, repetitions=1):
        for i in range(repetitions):
            result = self._dte.parmr([key])
            print(result)