From 7d1ec41f4ba1adfa904900b77444a3abaca5c7d9 Mon Sep 17 00:00:00 2001
From: Jordan Ruthe <jordan.ruthe@gmail.com>
Date: Sun, 31 Jan 2021 15:24:22 -0500
Subject: [PATCH] printer: Include quad_gantry_level

---
 ks_includes/KlipperScreen.conf | 7 +++++++
 ks_includes/printer.py         | 2 +-
 2 files changed, 8 insertions(+), 1 deletion(-)

diff --git a/ks_includes/KlipperScreen.conf b/ks_includes/KlipperScreen.conf
index e9eb78e6..f522bb9c 100644
--- a/ks_includes/KlipperScreen.conf
+++ b/ks_includes/KlipperScreen.conf
@@ -85,6 +85,13 @@ name: {{ gettext('Temperature') }}
 icon: heat-up
 panel: temperature
 
+[menu __main actions quad_gantry_level]
+name: {{ gettext('Quad Gantry Level') }}
+icon: home-z
+method: printer.gcode.script
+params: {"script":"QUAD_GANTRY_LEVEL"}
+enable: {{ printer.quad_gantry_level is defined }}
+
 [menu __main actions macros]
 name: {{ gettext('Macros') }}
 icon: custom-script
diff --git a/ks_includes/printer.py b/ks_includes/printer.py
index 9412bdb9..a7b8aafc 100644
--- a/ks_includes/printer.py
+++ b/ks_includes/printer.py
@@ -178,7 +178,7 @@ class Printer:
             }
         }
 
-        sections = ["bed_mesh","bltouch","probe"]
+        sections = ["bed_mesh","bltouch","probe","quad_gantry_level"]
         for section in sections:
             if self.config_section_exists(section):
                 data["printer"][section] = self.get_config_section(section).copy()