dashzero/main.py at master · panbachi/dashzero · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
import os
import yaml
import locale

from kivy.app import App
from kivy.clock import Clock
from kivy.core.window import Window
from kivy.uix.floatlayout import FloatLayout

from kivymd.app import MDApp

from gui.core import Navigation

from core.connectors import HomeAssistant
from core.platform import Shpi

class MainScreen(FloatLayout):
    def __init__(self, config, **kwargs):
        super(MainScreen, self).__init__(**kwargs)

        self.app = App.get_running_app()

        self.navigation = Navigation(config['rooms'])

        self.size = Window.size

        self.backlight_trigger = Clock.schedule_once(self.turn_display_off, 5)
        self.backlight = True

        self.add_widget(self.navigation)

    def on_touch_down(self, touch):
        if not self.backlight:
            print(touch)
            self.turn_display_on()
            print(self.backlight)

        Clock.unschedule(self.backlight_trigger)
        self.backlight_trigger = Clock.schedule_once(self.turn_display_off, 5)

        return super().on_touch_down(touch)

    def turn_display_off(self, dt):
        print('display is off')
        self.backlight = False

        if self.app.platform:
            self.app.platform.turn_display_off()

    def turn_display_on(self):
        print('display is on')
        self.backlight = True

        if self.app.platform:
            self.app.platform.turn_display_on()


class MainApp(MDApp):
    def __init__(self, **kwargs):
        self.register_event_type('on_state_changed')
        self.load_theme()

        super().__init__(**kwargs)
        self.config = config
        self.connectors = {}
        self.platform = None

        if 'homeassistant' in config:
            self.connectors['homeassistant'] = HomeAssistant(self, self.config['homeassistant'])

        if 'shpi' in config:
            self.platform = Shpi(self, self.config['shpi'])

    def build(self):
        self.root = MainScreen(config)
        Window.size = (800, 480)

        for connector in self.connectors:
            self.connectors[connector].connect_to_server()

        return self.root

    def change_state(self, connector, entity_id, state):
        if connector in self.connectors:
            self.connectors[connector].change_state(entity_id, state)

    def change_temperature(self, connector, entity_id, temperature):
        if connector in self.connectors:
            self.connectors[connector].change_temperature(entity_id, temperature)

    def on_state_changed(self, *args, **kwargs):
        pass

    def load_theme(self):
        theme_cfg = config.get('theme', {})
        self.theme_cls.theme_style = theme_cfg.get('style', 'Light')
        self.theme_cls.primary_palette = theme_cfg.get('primary_palette', 'Blue')
        self.theme_cls.accent_palette = theme_cfg.get('accent_palette', 'Amber')

def load_config():
    bin_path = os.path.dirname(__file__)
    config_file = os.path.join(bin_path, 'config.yaml')
    if not os.path.exists(config_file):
        config_file = os.path.join(bin_path, 'config.example.yaml')

    with open(config_file, 'r') as stream:
        config = yaml.safe_load(stream)

    return config

if __name__ == "__main__":
    global config
    config = load_config()

    # Use system defined locale
    locale.setlocale(locale.LC_ALL, '')

    MainApp().run()