From 93eb960a8ed4324bee6aaf3cb8719a3574912ce0 Mon Sep 17 00:00:00 2001
From: Robert Resch <robert@resch.dev>
Date: Wed, 27 Nov 2024 19:26:32 +0100
Subject: [PATCH] Don't override nickname

---
 bumper/db.py     | 3 ++-
 bumper/models.py | 3 ++-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/bumper/db.py b/bumper/db.py
index e868c7f..6b443f6 100644
--- a/bumper/db.py
+++ b/bumper/db.py
@@ -338,7 +338,8 @@ def revoke_expired_tokens() -> None:
 
 def bot_add(sn: str, did: str, dev_class: str, resource: str, company: str) -> None:
     """Add bot."""
-    new_bot = VacBotDevice()
+    bot = bot_get(did) or {}
+    new_bot = VacBotDevice(**bot)
     new_bot.did = did
     new_bot.name = sn
     new_bot.vac_bot_device_class = dev_class
diff --git a/bumper/models.py b/bumper/models.py
index 907835a..d00eec3 100644
--- a/bumper/models.py
+++ b/bumper/models.py
@@ -18,8 +18,9 @@ class VacBotDevice:
         name: str = "",
         nick: str = "",
         company: str = "",
+        **kwargs: Any,
     ):
-        self.vac_bot_device_class = vac_bot_device_class
+        self.vac_bot_device_class = vac_bot_device_class or kwargs.get("class", "")
         self.company = company
         self.did = did
         self.name = name