pystat/FileManager.py at master · willik/pystat · 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
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
__author__ = 'matt'

import threading
from ScheduleHelper import ScheduleEvent
from ThermostatConfiguration import ThermostatConfiguration

class FileManager():
    def __init__(self, configurationFileName, currentFileName, scheduleFileName):
        self.configurationFileName = configurationFileName
        self.currentFileName = currentFileName
        self.scheduleFileName = scheduleFileName
        self.configuration = None
        self.scheduleHelper = None

        self.configurationLock = threading.Lock()
        self.currentLock = threading.Lock()
        self.scheduleLock = threading.Lock()

    def read_configuration(self):
        self.configurationLock.acquire()
        configurationFile = open(self.configurationFileName)

        if configurationFile.readable():
            self.configuration = ThermostatConfiguration()
            for line in configurationFile:
                line = line.strip()
                if "" != line and "#" != line[0]:
                    pair = line.split()
                    if 1 < len(pair):
                        self.set_configuration_option(pair[0], pair[1], self.configuration)

        configurationFile.close()
        self.configurationLock.release()
        return self.configuration

    def read_current(self):
        self.currentLock.acquire()
        currentFile = open(self.currentFileName)
        current = None

        if currentFile.readable():
            current = {}
            for line in currentFile:
                line = line.strip()
                if "" != line and "#" != line[0]:
                    pair = line.split()
                    if 1 < len(pair) and ("mode" == pair[0] or "temperature" == pair[0]):
                        current[pair[0]] = pair[1]

        self.currentLock.release()
        currentFile.close()
        return current

    def write_current(self, mode, temperature):
        self.currentLock.acquire()
        # ensure values being written are valid
        if mode not in [ "off", "fan", "ac", "heat" ]:
            mode = "off"
        if temperature < self.configuration.minimumTemperature:
            temperature = self.configuration.minimumTemperature
        elif temperature > self.configuration.maximumTemperature:
            temperature = self.configuration.maximumTemperature

        currentFile = open(self.currentFileName, "w")

        if currentFile.writable():
            currentFile.write("mode " + str(mode) + "\ntemperature " + str(temperature))

        self.currentLock.release()
        currentFile.close()

    def read_schedule(self):
        scheduleFileContents = ""
        self.scheduleLock.acquire()

        scheduleFile = open(self.scheduleFileName)
        if scheduleFile.readable():
            scheduleFileContents = scheduleFile.read()

        self.scheduleLock.release()
        return scheduleFileContents

    def add_event(self, day, time, temperature):
        if self.scheduleHelper is not None:
            try:
                event = self.scheduleHelper.add_event(
                    ScheduleEvent(day, time, temperature))
                if event is None:
                    self.write_schedule(self.scheduleHelper.serialize_events())
                return event
            except:
                pass
        return None

    def delete_event(self, day, time):
        if self.scheduleHelper is not None:
            try:
                return self.scheduleHelper.delete_event(day, time)
            except:
                pass
        return None

    def write_schedule(self, scheduleContents):
        self.scheduleLock.acquire()

        scheduleFile = open(self.scheduleFileName, "w")
        if scheduleFile.writable():
            scheduleFile.write(scheduleContents)

        self.scheduleLock.release()

    def set_configuration_option(self, option_name, option_value, configuration):
        if "heat_pin" == option_name:
            configuration.heatPin = option_value
        elif "ac_pin" == option_name:
            configuration.acPin = option_value
        elif "fan_pin" == option_name:
            configuration.fanPin = option_value
        elif "active_hysteresis" == option_name:
            configuration.activeHysteresis = option_value
        elif "inactive_hysteresis" == option_name:
            configuration.inactiveHysteresis = option_value
        elif "email_alerts" == option_name:
            configuration.emailAlerts = ("true" == option_value.lower())
        elif "email" == option_name:
            configuration.email = option_value
        elif "error_threshold" == option_name:
            configuration.errorThreshold = option_value
        elif "weather_url" == option_name:
            configuration.weatherurl = option_value
        elif "weather_api_key" == option_name:
            configuration.weatherAPIKey = option_value
        elif "lat_long" == option_name:
            configuration.latlong = option_value