Add FreeBuds SE profile, close #17

melianmiko · Nov 27, 2023 · ee86e4d · ee86e4d
1 parent 3d3cf70
commit ee86e4d
Show file tree

Hide file tree

Showing 5 changed files with 18 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -19,6 +19,7 @@ Features:
 - Open source and free forever
 
 Supported devices:
+- HUAWEI FreeBuds SE
 - HUAWEI FreeBuds 4i
 - HUAWEI FreeLace Pro
 - HONOR Earbuds 2 Lite

diff --git a/docs/research.ods b/docs/research.ods
diff --git a/src/openfreebuds/device/huawei/__init__.py b/src/openfreebuds/device/huawei/__init__.py
@@ -1,15 +1,17 @@
-from openfreebuds.device.huawei.profiles import FreeLaceProDevice, FreeBuds4iDevice
+from openfreebuds.device.huawei.profiles import FreeLaceProDevice, FreeBuds4iDevice, FreeBudsSEDevice
 
 profiles = {
     "HUAWEI FreeBuds 4i": FreeBuds4iDevice,
     "HUAWEI FreeLace Pro": FreeLaceProDevice,
+    "HUAWEI FreeBuds SE" : FreeBudsSEDevice
 }
 
 # (implementation_level, profile)
 devices = {
     "HUAWEI FreeBuds 4i": ("full", FreeBuds4iDevice),
     "HONOR Earbuds 2 Lite": ("full", FreeBuds4iDevice),
     "HUAWEI FreeLace Pro": ("full", FreeLaceProDevice),
+    "HUAWEI FreeBuds SE": ("full", FreeBudsSEDevice),
 
     "HUAWEI FreeBuds 5i": ("partial", FreeBuds4iDevice),
     "HUAWEI FreeBuds Pro 2": ("partial", FreeBuds4iDevice),

diff --git a/src/openfreebuds/device/huawei/profiles.py b/src/openfreebuds/device/huawei/profiles.py
@@ -54,3 +54,15 @@ def __init__(self, address):
             TwsAutoPauseHandler(),
             VoiceLanguageHandler(),
         ]
+
+
+class FreeBudsSEDevice(GenericHuaweiSppDevice):
+    def __init__(self, address):
+        super().__init__(address)
+        self.handlers = [
+            DropLogsHandler(),
+            DeviceInfoHandler(),
+            BatteryHandler(),
+            DoubleTapConfigHandler(),
+            VoiceLanguageHandler(),
+        ]
diff --git a/src/openfreebuds_applet/ui/menu_device.py b/src/openfreebuds_applet/ui/menu_device.py
@@ -77,6 +77,8 @@ def __init__(self, manager: FreebudsManager):
     def on_build(self):
         device = self.manager.device
         current = device.find_property("anc", "mode", -1)
+        if current == -1:
+            return
         next_mode = (current + 1) % 3
 
         for a in range(0, 3):