Add work mode select (#449)

.github/workflows/ci.yml

@@ -52,3 +52,47 @@ jobs:
         run: mypy custom_components/
       - name: Pylint review
         run: pylint custom_components/
+
+  create_zip_file:
+    name: Create zip file for testing
+    permissions:
+      # Required to upload/save artifact, otherwise you'll get
+      # "Error: Resource not accessible by integration"
+      contents: write
+      # Required to post comment, otherwise you'll get
+      # "Error: Resource not accessible by integration"
+      pull-requests: write
+    needs:
+      - code-quality
+    runs-on: ubuntu-latest
+    if: github.event_name == 'pull_request'
+    steps:
+      - name: 📥 Checkout the repository
+        uses: actions/checkout@v4
+
+      - name: 📦 Create zip file for the integration
+        run: |
+          cd "${{ github.workspace }}/custom_components/deebot"
+          zip deebot.zip -r ./
+          mv deebot.zip "${{ github.workspace }}/"
+
+      - name: 📤 Create comment with the artifiactory
+        uses: gavv/pull-request-artifacts@v2
+        with:
+          # Commit hash that triggered PR
+          commit: ${{ github.event.pull_request.head.sha }}
+
+          # Token for current repo (used to post PR comment)
+          repo-token: ${{ secrets.GITHUB_TOKEN }}
+
+          artifacts-branch: artifacts
+          inter-link: false
+
+          comment-title: |
+            🤖 Pull request artifactsy
+
+            For testing please download the zip file below and extract it in `[config folder]/custom_components/deebot`.
+
+          # Whitespace-separated list of files to upload
+          artifacts: |
+            deebot.zip

custom_components/deebot/manifest.json

@@ -14,7 +14,7 @@
     "deebot_client"
   ],
   "requirements": [
-    "git+https://github.com/DeebotUniverse/client.py@dev#deebot-client==4.0.0b0",
+    "git+https://github.com/DeebotUniverse/client.py@dev#deebot-client==4.0.0dev1",
     "numpy>=1.23.2"
   ],
   "version": "v0.0.0"

custom_components/deebot/select.py

@@ -44,6 +44,16 @@ class DeebotSelectEntityDescription(
         icon="mdi:water",
         entity_category=EntityCategory.CONFIG,
     ),
+    DeebotSelectEntityDescription(
+        capability_fn=lambda caps: caps.clean.work_mode,
+        current_option_fn=lambda e: e.mode.display_name,
+        options_fn=lambda cap: [mode.display_name for mode in cap.types],
+        key="work_mode",
+        translation_key="work_mode",
+        entity_registry_enabled_default=False,
+        icon="mdi:cog",
+        entity_category=EntityCategory.CONFIG,
+    ),
 )
 
 

custom_components/deebot/translations/en.json

@@ -81,6 +81,15 @@
           "medium": "medium",
           "ultrahigh": "ultrahigh"
         }
+      },
+      "work_mode": {
+        "name": "Work mode",
+        "state": {
+          "mop": "Mop",
+          "mop_after_vacuum": "Mop after vacuum",
+          "vacuum": "Vacuum",
+          "vacuum_and_mop": "Vacuum & mop"
+        }
       }
     },
     "sensor": {

requirements.txt

@@ -1,5 +1,5 @@
 # if code from specific branch is needed
-git+https://github.com/DeebotUniverse/client.py@dev#deebot-client==4.0.0b0
+git+https://github.com/DeebotUniverse/client.py@dev#deebot-client==4.0.0dev1
 #deebot-client==3.0.2
 
 homeassistant>=2023.8.0b0

translations.schema.json

@@ -213,6 +213,33 @@
                 }
               },
               "type": "object"
+            },
+            "work_mode": {
+              "additionalProperties": false,
+              "properties": {
+                "name": {
+                  "type": "string"
+                },
+                "state": {
+                  "additionalProperties": false,
+                  "properties": {
+                    "mop": {
+                      "type": "string"
+                    },
+                    "mop_after_vacuum": {
+                      "type": "string"
+                    },
+                    "vacuum": {
+                      "type": "string"
+                    },
+                    "vacuum_and_mop": {
+                      "type": "string"
+                    }
+                  },
+                  "type": "object"
+                }
+              },
+              "type": "object"
             }
           },
           "type": "object"