from .models import *
from ospaz_site.settings import DEBUG
from threading import Thread
import time
from datetime import datetime, timedelta
from pyModbusTCP.client import ModbusClient


class MbClearHistoryService(Thread):
    def __init__(self, model, save_days):
        super().__init__()
        self.model = model
        self._days = save_days

    def run(self):
        while True:
            self.model.objects.filter(dt__lt=(datetime.now() - timedelta(days=60))).delete()
            time.sleep(10)


class MbService(Thread):
    def __init__(self, ip_addr, port, scan_rate, poll_time_ms=60000, save_days=60):
        super().__init__()
        self._poll_time_ms = poll_time_ms
        self._scan_rate = scan_rate

        self.mb = ModbusClient(host=ip_addr, port=port, debug=DEBUG, auto_open=True)

    def _init_state(self):
        """
            Функция вызывается один раз при создании сервиса, в ней нужно загрузить содержимое регистров из базы данных.
        """
        pass

    def _push_current_state(self):
        """
            Функция вызывается в момент, когда требуется сохранить состояние в базу данных
        """
        pass

    def _load_current_state(self):
        """
            Функция должна совершить операцию чтения регистров modbus
        """
        pass

    def _check_need_save(self):
        """
            Функция должна вернуть результат: нужно ли сохранять текущее состояние в базу данных
        """
        return True

    def run(self):
        # if self.log_type == 'on-change':
        #     pass
        # else:
        print(f"[{datetime.now().isoformat()}] MbService:{type(self).__name__}: service started!")
        self._init_state()

        last_query = datetime.now()
        scan_rate = timedelta(milliseconds=self._scan_rate)

        while True:
            try:
                self._load_current_state()
                if self._check_need_save():
                    self._push_current_state()
            except Exception as ex:
                print(f"[{datetime.now().isoformat()}] MbService:{type(self).__name__}: Exception: {ex}")

            # вычислим время до следующего опроса и подождем
            # время следующего опроса
            need_time = last_query + scan_rate
            curr_time = datetime.now()
            if need_time > curr_time:
                delta = need_time - curr_time
                time.sleep(delta.seconds + (delta.microseconds / 1000000))
                last_query = need_time
            else:
                last_query = datetime.now()


class MbTankService(MbService):
    def __init__(self, **kwargs):
        super().__init__(**kwargs)

        self._curr_state = None
        # {
        #      "level": int,
        #      "status": int,
        #      "radar": int
        # }

    def _init_state(self):
        pass

    def _push_current_state(self):
        record = MbTankRecord(radar_raw=self._curr_state['radar'],
                              level=self._curr_state['level'],
                              status=self._curr_state['status'])
        record.save()

    def _load_current_state(self):
        # D0: level_percentage
        # D5: status_reg
        # D1: radar_high_reg
        # D2: radar_low_reg
        values = self.mb.read_holding_registers(0, 6)
        self._curr_state = {
            "level": values[0],
            "status": values[5],
            "radar": (values[1] << 16) | values[2]
        }

    def _check_need_save(self):
        return True


class MbPumpService(MbService):
    _config = {
        "modbus": {
            "host": "10.8.105.2",
            "port": 503,
            "log-type": "on-change"
        },
        "registers": [
            ["D0", "v231_v236"],
            ["D1", "v236_v29"],
            ["D15", "ps_39"],
            ["D16", "flow_meter"],
            ["D28", "pump_stage"],
            ["D30", "half_auto_control"],
            ["D31", "watch_1"],
            ["D32", "watch_2"],

            ["D33", "st_valves"],
            ["D35", "alarms"],

            ["D25", "vfd_freq"],
            ["D26", "vfd_current"],
            ["D27", "vfd_error"]
        ]
    }