io_data.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-

# Copyright (C)2018-24 Maurizio Montel (dr-prodigy) <maurizio.montel@gmail.com>
# This file is part of hompi <https://github.com/dr-prodigy/hompi>.
#
# hompi is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# hompi is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with hompi.  If not, see <http://www.gnu.org/licenses/>.

import json
import datetime

import config

hide_message_time = None


class Input:
    def __init__(self):
        self.last_update = datetime.datetime.now().isoformat()
        self.data = ''


class Status:
    def __init__(self):
        global hide_message_time
        # general
        self.id = config.HOMPI_ID
        self.last_update = datetime.datetime.now().isoformat()
        self.last_change = datetime.datetime.now().isoformat()
        self.hompi_slaves = {}
        self.hompi_ext_sensors = {}
        # heating
        if config.MODULE_TEMP:
            self.int_temp_c = 0.0
        # TRV
        if config.MODULE_TRV:
            self.areas = {}
            self.req_area_temps = ''
        self.mode_id = ''
        self.mode_desc = ''
        self.short_mode_desc = ''
        self.timetable_desc = ''
        self.day_type_desc = ''
        self.req_temp_c = 0.0
        self.req_temp_desc = ''
        self.req_start_time = 0
        self.req_end_time = 0
        self.heating_status = 'off'
        # switches
        self.sw_sig = [False]*len(config.BUTTONS)
        self.sw_status = [False]*len(config.BUTTONS)
        # gui
        self.current_image = ''
        self.message = ''
        hide_message_time = datetime.datetime.now()
        # ambient
        if config.MODULE_AMBIENT:
            self.ambient_color = '000000'
            self.ambient_effect = None
            self.ambient_on = False
        # meteo
        if config.MODULE_METEO:
            self.ext_temp_c = 0.0
            self.place = ''
            self.weather = ''
            self.humidity = 0
            self.pressure = 0
            self.wind = 0.0
        # aphorism
        if config.MODULE_APHORISM:
            self.aphorism_text = ''
            self.aphorism_author = ''

    def get_output(self):
        return json.dumps(self.__dict__, indent=0)

    def get_status_text(self):
        if self.message != '':
            status = '{}'.format(self.message)
        else:
            state = self.mode_desc.lower().replace('-', ' ')
            heating = self.heating_status.lower().replace('-', ' ')
            status = 'temperature {:.1f} degrees, {} mode'.\
                     format(self.int_temp_c, state) + \
                     (', target {:.1f} degrees, heating {}'.format(
                        self.req_temp_c, heating)
                         if state in ['automatic', 'manual'] else
                         '')
        return status

    def set_ambient(self, ambient):
        self.ambient_color = ambient.status_color
        self.ambient_on = ambient.status_power_on
        self.ambient_effect = ambient.status_effect

    def update(self, current_time):
        if self.message != '' and hide_message_time <= current_time:
            self.message = ''
            self.last_change == current_time.isoformat()

    def send_message(self, message):
        global hide_message_time

        self.message = message.encode('utf-8')
        self.last_change = datetime.datetime.now().isoformat()
        hide_message_time = datetime.datetime.now() + datetime.timedelta(seconds=40)

    def reset_message(self):
        global hide_message_time

        self.message = ''
        self.last_change = datetime.datetime.now().isoformat()
        hide_message_time = datetime.datetime.now()

    def send_switch_command(self, switch):
        self.sw_sig[switch] = True

    def get_area_manual_set(self):
        is_manual_set = False
        if config.MODULE_TRV:
            for area in self.areas.values():
                if 'manual_set' in area:
                    is_manual_set |= area['manual_set']
        return is_manual_set

class SystemInfo:
    def __init__(self):
        self.modules = []
        self.buttons = []
        self.temperatures = []

        if config.MODULE_AMBIENT:
            self.modules.append('ambient')
            self.ambient_commands = ['AMBIENT_XMAS']
        if config.MODULE_APHORISM:
            self.modules.append('aphorism')
        if config.MODULE_DB_LOG:
            self.modules.append('db_log')
        if config.MODULE_LCD:
            self.modules.append('lcd')
        if config.MODULE_METEO:
            self.modules.append('meteo')
        if config.MODULE_TEMP:
            self.modules.append('temp')
        for button in config.BUTTONS:
            self.buttons.append(button[1])

    def get_output(self):
        return json.dumps(self.__dict__, indent=0)