1 files changed, 10 insertions, 0 deletions
diff --git a/lorrycontroller/statedb.py b/lorrycontroller/statedb.py
index 8c62a31..99ea7fc 100644
--- a/lorrycontroller/statedb.py
+++ b/lorrycontroller/statedb.py
@@ -440,6 +440,16 @@ class StateDB(object):
             'UPDATE lorries SET last_run=? WHERE path=?',
             (last_run, path))
 
+    def set_lorry_last_run_exit_and_output(self, path, exit, output):
+        logging.debug(
+            'StateDB.set_lorry_last_run_exit_and_output(%r, %r, %r) called',
+            path, exit, output)
+        assert self.in_transaction
+        c = self.get_cursor()
+        c.execute(
+            'UPDATE lorries SET last_run_exit=?, last_run_error=? WHERE path=?',
+            (exit, output, path))
+
     def set_lorry_disk_usage(self, path, disk_usage):
         logging.debug(
             'StateDB.set_lorry_disk_usage(%r, %r) called', path, disk_usage)