blob: 7fe4bfeda66a77e34ea4d2693bac88775ce545ad (
plain)
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
|
# Add ability to define custom g-code macros
#
# Copyright (C) 2018 Kevin O'Connor <kevin@koconnor.net>
#
# This file may be distributed under the terms of the GNU GPLv3 license.
import logging
DEFAULT_PREFIX = 'default_parameter_'
class GCodeMacro:
def __init__(self, config):
self.alias = config.get_name().split()[1].upper()
self.script = config.get('gcode')
printer = config.get_printer()
self.gcode = printer.lookup_object('gcode')
try:
self.gcode.register_command(
self.alias, self.cmd, desc=self.cmd_desc)
except self.gcode.error as e:
raise config.error(str(e))
self.in_script = False
self.kwparams = { o[len(DEFAULT_PREFIX):].upper(): config.get(o)
for o in config.get_prefix_options(DEFAULT_PREFIX) }
cmd_desc = "G-Code macro"
def cmd(self, params):
if self.in_script:
raise self.gcode.error(
"Macro %s called recursively" % (self.alias,))
script = ""
kwparams = dict(self.kwparams)
kwparams.update(params)
try:
script = self.script.format(**kwparams)
except Exception:
msg = "Macro %s script formatting failed" % (self.alias,)
logging.exception(msg)
raise self.gcode.error(msg)
self.in_script = True
try:
self.gcode.run_script_from_command(script)
finally:
self.in_script = False
def load_config_prefix(config):
return GCodeMacro(config)
|
