From b99ed631c567af91eeb4b6fbd65fb031aa95a8b6 Mon Sep 17 00:00:00 2001
From: Ludeeus <ludeeus@ludeeus.dev>
Date: Sat, 5 Dec 2020 12:44:17 +0000
Subject: [PATCH] Add data to job

---
 supervisor/jobs/__init__.py | 16 ++++++++++++++--
 1 file changed, 14 insertions(+), 2 deletions(-)

diff --git a/supervisor/jobs/__init__.py b/supervisor/jobs/__init__.py
index c84b541e2..f480e2509 100644
--- a/supervisor/jobs/__init__.py
+++ b/supervisor/jobs/__init__.py
@@ -22,6 +22,7 @@ class SupervisorJob(CoreSysAttributes):
         self._name: Optional[str] = name
         self._progress: int = 0
         self._stage: Optional[str] = None
+        self._data: Optional[dict] = None
 
     @property
     def id(self) -> str:
@@ -43,8 +44,16 @@ class SupervisorJob(CoreSysAttributes):
         """Return the current stage."""
         return self._stage
 
+    @property
+    def data(self) -> Optional[str]:
+        """Return additional data for the job."""
+        return self._data
+
     def update(
-        self, progress: Optional[int] = None, stage: Optional[str] = None
+        self,
+        progress: Optional[int] = None,
+        stage: Optional[str] = None,
+        data: Optional[dict] = None,
     ) -> None:
         """Update the job object."""
         if progress is not None:
@@ -54,11 +63,14 @@ class SupervisorJob(CoreSysAttributes):
             self._progress = round(progress)
         if stage is not None:
             self._stage = stage
+        if data is not None:
+            self._data = data
         _LOGGER.debug(
-            "Job updated; name: %s, progress: %s, stage: %s",
+            'Update: {"name": %s, "progress": %s, "stage": %s, "data": %s}',
             self.name,
             self.progress,
             self.stage,
+            self.data,
         )