[#88650] server: src: device_mgmt: Update device version

Signed-off-by: Anna Roszkiewicz <[email protected]>

Author: anroszkiewicz

common/communication/src/request_models.py

@@ -103,6 +103,11 @@ class UpdateProgress(Request):
     progress: int
 
 
+class UpdateVersion(Request):
+    method: Literal['update_version'] = 'update_version'
+    version: str
+
+
 class Container(BaseModel):
     """container holds a list of models to enable parsing from json"""
     data: Union[
@@ -118,4 +123,5 @@ class Container(BaseModel):
         FsFileProbe,
         FsFileProbeReply,
         UpdateProgress,
+        UpdateVersion,
     ]

server/src/device_mgmt/models/remote_device.py

@@ -14,6 +14,7 @@
     FsFileDownloadReply,
     FsFileProbeReply,
     UpdateProgress,
+    UpdateVersion,
 )
 
 import simple_websocket
@@ -24,6 +25,7 @@
     RDFM_WS_MISSING_CAPABILITIES,
     WebSocketException,
 )
+from rdfm.schema.v1.updates import META_SOFT_VER
 import device_mgmt.action
 import server
 
@@ -184,6 +186,12 @@ def __handle_device_message(self, request: Request):
                     f"Device {self.token.device_id} update in progress: {request.progress}%",
                     flush=True,
                 )
+        elif isinstance(request, UpdateVersion):
+            device = server.instance._devices_db.get_device_data(self.token.device_id)
+            metadata = json.loads(device.device_metadata)
+            metadata[META_SOFT_VER] = request.version
+            server.instance._devices_db.update_metadata(self.token.device_id, metadata)
+
         else:
             print("Unknown request:", request, flush=True)
             raise WebSocketException(