mirror of
https://github.com/QIDITECH/klipper.git
synced 2026-01-31 07:58:42 +03:00
123 lines
4.5 KiB
Python
123 lines
4.5 KiB
Python
# Save arbitrary variables so that values can be kept across restarts.
|
|
#
|
|
# Copyright (C) 2020 Dushyant Ahuja <dusht.ahuja@gmail.com>
|
|
# Copyright (C) 2016-2020 Kevin O'Connor <kevin@koconnor.net>
|
|
#
|
|
# This file may be distributed under the terms of the GNU GPLv3 license.
|
|
import os, logging, ast, configparser
|
|
|
|
class SaveVariables:
|
|
def __init__(self, config):
|
|
self.printer = config.get_printer()
|
|
self.filename = os.path.expanduser(config.get('filename'))
|
|
self.allVariables = {}
|
|
|
|
self.default_values = {
|
|
'auto_reload_detect': 0, 'auto_read_rfid': 0, 'auto_init_detect': 0,
|
|
'box_count': 0, 'enable_box': 0,'load_retry_num': 0,
|
|
'slot_sync': "", 'retry_step': "", 'last_load_slot': ""
|
|
}
|
|
|
|
for i in range(16):
|
|
self.default_values[f'filament_slot{i}'] = 0
|
|
self.default_values[f'color_slot{i}'] = 0
|
|
self.default_values[f'vendor_slot{i}'] = 0
|
|
self.default_values[f'slot{i}'] = 0
|
|
self.default_values[f'value_t{i}'] = ""
|
|
self.default_values[f'runout_{i}'] = 0
|
|
|
|
self.default_values['filament_slot16'] = 0
|
|
self.default_values['color_slot16'] = 0
|
|
self.default_values['vendor_slot16'] = 0
|
|
|
|
for key, value in self.default_values.items():
|
|
setattr(self, key, value)
|
|
|
|
try:
|
|
if not os.path.exists(self.filename):
|
|
open(self.filename, "w").close()
|
|
self.loadVariables()
|
|
except self.printer.command_error as e:
|
|
raise config.error(str(e))
|
|
gcode = self.printer.lookup_object('gcode')
|
|
gcode.register_command('SAVE_VARIABLE', self.cmd_SAVE_VARIABLE,
|
|
desc=self.cmd_SAVE_VARIABLE_help)
|
|
|
|
def load_variable(self, section, option):
|
|
varfile = configparser.ConfigParser()
|
|
try:
|
|
varfile.read(self.filename)
|
|
return varfile.get(section, option)
|
|
except:
|
|
msg = "Unable to parse existing variable file"
|
|
logging.exception(msg)
|
|
raise self.printer.command_error(msg)
|
|
|
|
def save_variable(self, section, option, value):
|
|
varfile = configparser.ConfigParser()
|
|
try:
|
|
varfile.read(self.filename)
|
|
if not varfile.has_section(section):
|
|
varfile.add_section(section)
|
|
varfile.set(section, option, value)
|
|
with open(self.filename, 'w') as configfile:
|
|
varfile.write(configfile)
|
|
except Exception as e:
|
|
msg = "Unable to save variable"
|
|
logging.exception(msg)
|
|
raise self.printer.command_error(msg)
|
|
self.loadVariables()
|
|
|
|
def loadVariables(self):
|
|
allvars = {}
|
|
varfile = configparser.ConfigParser()
|
|
try:
|
|
varfile.read(self.filename)
|
|
if varfile.has_section('Variables'):
|
|
for name, val in varfile.items('Variables'):
|
|
allvars[name] = ast.literal_eval(val)
|
|
except:
|
|
msg = "Unable to parse existing variable file"
|
|
logging.exception(msg)
|
|
raise self.printer.command_error(msg)
|
|
self.allVariables = allvars
|
|
|
|
for key, default in self.default_values.items():
|
|
setattr(self, key, self.allVariables.get(key, default))
|
|
|
|
cmd_SAVE_VARIABLE_help = "Save arbitrary variables to disk"
|
|
|
|
def cmd_SAVE_VARIABLE(self, gcmd):
|
|
varname = gcmd.get('VARIABLE')
|
|
value = gcmd.get('VALUE')
|
|
try:
|
|
value = ast.literal_eval(value)
|
|
except ValueError as e:
|
|
raise gcmd.error("Unable to parse '%s' as a literal" % (value,))
|
|
newvars = dict(self.allVariables)
|
|
newvars[varname] = value
|
|
# Write file
|
|
varfile = configparser.ConfigParser()
|
|
varfile.add_section('Variables')
|
|
for name, val in sorted(newvars.items()):
|
|
varfile.set('Variables', name, repr(val))
|
|
try:
|
|
f = open(self.filename, "w")
|
|
varfile.write(f)
|
|
f.close()
|
|
except:
|
|
msg = "Unable to save variable"
|
|
logging.exception(msg)
|
|
raise gcmd.error(msg)
|
|
self.loadVariables()
|
|
|
|
def get_status(self, eventtime):
|
|
status = {'variables': self.allVariables}
|
|
|
|
for key in self.default_values.keys():
|
|
status[key] = getattr(self, key)
|
|
|
|
return status
|
|
|
|
def load_config(config):
|
|
return SaveVariables(config) |