MixwareScreen/config.py at main · Mixwarebot/MixwareScreen · 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
135
136
137
import logging
import os
import platform

from qtCore import *


class MixwareScreenConfig:
    config = None
    default_file = "/default_config.ini"
    file = "/.config.ini"
    format = QSettings.IniFormat
    width = 400
    height = 1280
    theme = "Light"
    language = "English"
    folder_rootPath = "~/printer_data/gcodes/"

    def __init__(self, path: str):
        self.path = path

        # Automatically increment the version number during testing
        self.default_config = QSettings(str(self.path + self.default_file), self.format)
        self.latest_version = self.default_config.value('app/version')
        self.latest_version_array = str(self.latest_version).split('.')
        if not is_release:
            self.default_config.setValue('app/version',
                                         f"{self.latest_version_array[0]}."
                                         f"{self.latest_version_array[1]}."
                                         f"{self.latest_version_array[2]}."
                                         f"{int(self.latest_version_array[3]) + 1}")
            self.default_config.sync()

        # Create a local config.ini if it does not exist
        if not os.path.isfile(self.path + self.file):
            self.reset_local_config()

        self.config = QSettings(str(self.path + self.file), self.format)

        # Check the latest version
        if self.get_version() != self.latest_version:
            self.set_version(self.latest_version)

        self.width = int(self.config.value('window/width'))
        self.height = int(self.config.value('window/height'))
        self.theme = self.config.value('window/theme')
        self.language = self.config.value('window/language')
        self.folder_rootPath = self.config.value('folder/root')

        if int(self.latest_version_array[1]) == 1 and int(self.latest_version_array[3]) > 106:
            if self.folder_rootPath == '/home/hyper-x/printer_data/gcodes':
                self.folder_rootPath = '/home/hyper-x/printer_data'
                self.config.setValue('folder/root', self.folder_rootPath)

        self._should_show_welcome = False
        try:
            self._should_show_welcome = int(self.config.value('window/welcome')) == 1
        except:
            os.system('sudo clear')
            self.set_value('window/welcome', 1)
            self._should_show_welcome = True
            logging.error("Related configuration not found, restart")
            os.system("sudo systemctl restart MixwareScreen.service")

        self._enable_power_loss_recovery = False
        try:
            self._enable_power_loss_recovery = int(self.config.value('window/power_loss_recovery')) == 1
        except:
            os.system('sudo clear')
            self.set_enable_power_loss_recovery(1)
            logging.error("Related configuration not found, restart")
            os.system("sudo systemctl restart MixwareScreen.service")

    def set_value(self, key: str, value):
        self.config.setValue(key, value)
        self.config.sync()

    def get_config(self):
        return self.config

    @property
    def should_show_welcome(self):
        return self._should_show_welcome

    def disable_show_welcome(self):
        self.set_value('window/welcome', 0)

    def get_window_size(self):
        return QSize(self.width, self.height)

    def get_width(self):
        return self.width

    def get_height(self):
        return self.height

    def set_theme(self, theme: str):
        self.set_value('window/theme', theme)

    def get_theme(self):
        return self.config.value('window/theme')

    def set_language(self, lang: str):
        self.set_value('window/language', lang)

    def get_language(self):
        return self.config.value('window/language')

    def get_folder_rootPath(self):
        return self.folder_rootPath

    def get_version(self):
        return self.config.value('app/version')

    def set_version(self, version: str):
        self.set_value('app/version', version)

    def reset_local_config(self):
        with open(self.path + self.default_file, 'r') as src:
            with open(self.path + self.file, 'w') as dest:
                dest.write(src.read())

    def enable_power_loss_recovery(self):
        return self._enable_power_loss_recovery

    def set_enable_power_loss_recovery(self, value: [0, 1]):
        self.set_value('window/power_loss_recovery', value)
        self._enable_power_loss_recovery = value == 1

    def get_power_loss_path(self):
        path = self.config.value('folder/power_loss_path')
        if not path:
            path = ""
        return path

    def set_power_loss_path(self, path: str):
        self.set_value('folder/power_loss_path', path)